Update structformer.py

structformer.py

@@ -356,7 +356,7 @@ class Transformer(nn.Module):
       self.pos_emb = nn.Embedding(500, hidden_size)
 
     self.layers = nn.ModuleList([
-        layers.TransformerLayer(hidden_size, nhead, hidden_size * 4, dropout,
+        TransformerLayer(hidden_size, nhead, hidden_size * 4, dropout,
                                 dropatt=dropatt, relative_bias=relative_bias)
         for _ in range(nlayers)])
 
@@ -474,12 +474,12 @@ class StructFormer(Transformer):
         pad=pad)
 
     self.parser_layers = nn.ModuleList([
-        nn.Sequential(layers.Conv1d(hidden_size, conv_size),
+        nn.Sequential(Conv1d(hidden_size, conv_size),
                       nn.LayerNorm(hidden_size, elementwise_affine=False),
                       nn.Tanh()) for i in range(n_parser_layers)])
 
     self.distance_ff = nn.Sequential(
-        layers.Conv1d(hidden_size, 2),
+        Conv1d(hidden_size, 2),
         nn.LayerNorm(hidden_size, elementwise_affine=False), nn.Tanh(),
         nn.Linear(hidden_size, 1))