Update deepscreen/models/predictors/graph_dta.py

deepscreen/models/predictors/graph_dta.py

@@ -22,7 +22,7 @@ class GraphDTA(LightningModule):
 
         # protein sequence encoder (1d conv)
         self.embedding_xt = nn.Embedding(num_features_protein, embed_dim)
-        self.conv_xt = nn.Conv1d(in_channels=1000, out_channels=n_filters, kernel_size=8)
+        self.conv_xt = nn.LazyConv1d(in_channels=1000, out_channels=n_filters, kernel_size=8)
         self.fc1_xt = nn.Linear(32 * 121, output_dim)
 
         # combined layers