Spaces:

ChemFM
/

reaction_prediction

Sleeping

feiyang-cai commited on 16 days ago

Commit

11b6a1c

verified ·

1 Parent(s): 954342d

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -17,7 +17,7 @@ from rdkit import RDLogger, Chem
 RDLogger.DisableLog('rdApp.*')
 DEFAULT_PAD_TOKEN = "[PAD]"
-device_map = "cpu"
 def compute_rank(prediction,raw=False,alpha=1.0):
     valid_score = [[k for k in range(len(prediction[j]))] for j in range(len(prediction))]

 RDLogger.DisableLog('rdApp.*')
 DEFAULT_PAD_TOKEN = "[PAD]"
+device_map = "cuda"
 def compute_rank(prediction,raw=False,alpha=1.0):
     valid_score = [[k for k in range(len(prediction[j]))] for j in range(len(prediction))]