mgyigit commited on
Commit
ed0dbd2
·
verified ·
1 Parent(s): bb27c74

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +5 -2
inference.py CHANGED
@@ -89,8 +89,11 @@ class Inference(object):
89
  bond_encoder=self.bond_encoder,
90
  bond_decoder=self.bond_decoder)
91
 
92
- self.sample_num = config.sample_num if config.sample_num < self.inf_dataset.num_smiles else self.inf_dataset.num_smiles
93
-
 
 
 
94
  self.inf_loader = DataLoader(self.inf_dataset,
95
  shuffle=True,
96
  batch_size=self.inf_batch_size,
 
89
  bond_encoder=self.bond_encoder,
90
  bond_decoder=self.bond_decoder)
91
 
92
+ try:
93
+ self.sample_num = config.sample_num if config.sample_num < self.inf_dataset.num_smiles else self.inf_dataset.num_smiles
94
+ except:
95
+ self.sample_num = config.sample_num
96
+
97
  self.inf_loader = DataLoader(self.inf_dataset,
98
  shuffle=True,
99
  batch_size=self.inf_batch_size,