Merge pull request #6 from woo1/test

remove duplicated code
microsoft · Dec 15, 2021 · 1c489e3 · 1c489e3
2 parents baa8022 + 98ee6b0
commit 1c489e3
Showing 1 changed file with 1 addition and 4 deletions.
diff --git 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)