feiyang-cai commited on
Commit
2a0c1ae
·
verified ·
1 Parent(s): 8f4c1fe

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -1
utils.py CHANGED
@@ -240,7 +240,8 @@ class ReactionPredictionModel():
240
  pad_token_id=self.tokenizer.pad_token_id,
241
  length_penalty=0.0,
242
  )
243
-
 
244
  original_smiles_list = self.tokenizer.batch_decode(outputs[:, len(inputs['input_ids'][0]):],
245
  skip_special_tokens=True)
246
  original_smiles_list = map(lambda x: x.replace(" ", ""), original_smiles_list)
 
240
  pad_token_id=self.tokenizer.pad_token_id,
241
  length_penalty=0.0,
242
  )
243
+
244
+ print(outputs)
245
  original_smiles_list = self.tokenizer.batch_decode(outputs[:, len(inputs['input_ids'][0]):],
246
  skip_special_tokens=True)
247
  original_smiles_list = map(lambda x: x.replace(" ", ""), original_smiles_list)