Update modeling_gcn.py

modeling_gcn.py

@@ -3,7 +3,7 @@ import torch.nn as nn
 import torch.nn.functional as F
 from torch_scatter import scatter
 from transformers import PreTrainedModel
-from configuration_gcn import GCNConfig
+# from configuration_gcn import GCNConfig
 import torch
 from rdkit import Chem
 from rdkit.Chem import AllChem
@@ -202,11 +202,14 @@ class GCNModel(PreTrainedModel):
 
 
 if __name__ == "__main__":
-    gcn_config = GCNConfig.from_pretrained("custom-gcn")
-
-    gcnd = GCNModel(gcn_config)
-    gcnd.model.load_state_dict(torch.load(r'G:\Trans_MXM\gcn_model\gcn.pt'))
-    gcnd.save_pretrained("custom-gcn")
+    pass
+    # gcn_config = GCNConfig(input_feature=64, emb_input=20, hidden_size=64, n_layers=6, num_classes=1, smiles=["C", "CC", "CCC"], processor_class="SmilesProcessor")
+    # gcn_config.save_pretrained("custom-gcn")
+    # gcn_config = GCNConfig.from_pretrained("custom-gcn")
+
+    # gcnd = GCNModel(gcn_config)
+    # gcnd.model.load_state_dict(torch.load(r'G:\Trans_MXM\gcn_model\gcn.pt'))
+    # gcnd.save_pretrained("custom-gcn")
 
     # gcnd1 = GCNModelForMolecularPrediction(gcn_config)
     #