Huhujingjing
/

custom-transmxm

Feature Extraction

Model card Files Files and versions Community

Huhujingjing commited on Jul 2, 2023

Commit

4944fb9

·

1 Parent(s): 1187aa4

Upload model

Files changed (1) hide show

modeling_transmxm.py +1 -1

modeling_transmxm.py CHANGED Viewed

@@ -1228,7 +1228,7 @@ class TransmxmModel(PreTrainedModel):
         self.pred_data = None
     def forward(self, tensor):
-        return self.bhackbone.forward_features(tensor)
     def SmilesProcessor(self, smiles):
         return self.process.get_data(smiles)

         self.pred_data = None
     def forward(self, tensor):
+        return self.backbone.forward_features(tensor)
     def SmilesProcessor(self, smiles):
         return self.process.get_data(smiles)