From 98ee6b0893058fb5661dea18c20b412e8f4b97f5 Mon Sep 17 00:00:00 2001 From: woo1 Date: Fri, 15 Oct 2021 10:51:51 +0900 Subject: [PATCH] remove duplicated code --- src/modeling/bert/modeling_graphormer.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/src/modeling/bert/modeling_graphormer.py b/src/modeling/bert/modeling_graphormer.py index 8acc057..255caee 100644 --- a/src/modeling/bert/modeling_graphormer.py +++ b/src/modeling/bert/modeling_graphormer.py @@ -126,10 +126,7 @@ def __init__(self, config): self.mesh_type = config.mesh_type if self.has_graph_conv == True: - if self.mesh_type=='hand': - self.graph_conv = GraphResBlock(config.hidden_size, config.hidden_size, mesh_type=self.mesh_type) - elif self.mesh_type=='body': - self.graph_conv = GraphResBlock(config.hidden_size, config.hidden_size, mesh_type=self.mesh_type) + self.graph_conv = GraphResBlock(config.hidden_size, config.hidden_size, mesh_type=self.mesh_type) self.intermediate = BertIntermediate(config) self.output = BertOutput(config)