Spaces:
Running
Running
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]
|