Spaces:
Sleeping
Sleeping
Update trainer.py
Browse files- 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]
|