osbm commited on
Commit
445a3c5
·
1 Parent(s): ccba61f

Update trainer.py

Browse files
Files changed (1) hide show
  1. trainer.py +1 -1
trainer.py CHANGED
@@ -704,7 +704,7 @@ class Trainer(object):
704
  G2_path = os.path.join(self.inference_model, '{}-G2.ckpt'.format(self.submodel))
705
  self.G2.load_state_dict(torch.load(G2_path, map_location=lambda storage, loc: storage))
706
 
707
-
708
  drug_smiles = [line for line in open("data/akt_test.smi", 'r').read().splitlines()]
709
 
710
  drug_mols = [Chem.MolFromSmiles(smi) for smi in drug_smiles]
 
704
  G2_path = os.path.join(self.inference_model, '{}-G2.ckpt'.format(self.submodel))
705
  self.G2.load_state_dict(torch.load(G2_path, map_location=lambda storage, loc: storage))
706
 
707
+ print(G_path)
708
  drug_smiles = [line for line in open("data/akt_test.smi", 'r').read().splitlines()]
709
 
710
  drug_mols = [Chem.MolFromSmiles(smi) for smi in drug_smiles]