molecular_property_prediction_zero_gpu

Running on Zero

feiyang-cai commited on 1 day ago

Commit

96a5b46

verified ·

1 Parent(s): 2dc971f

add cuda

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -229,7 +229,7 @@ class MolecularPropertyPredictionModel():
                 self.apapter_scaler_path[adapter_name] = None
                 assert dataset_task_types[adapter_name] == "classification", f"{adapter_name} is not a regression task."
-        #self.base_model.to("cuda")
         #print(self.base_model)
     def swith_adapter(self, adapter_name, adapter_id):

                 self.apapter_scaler_path[adapter_name] = None
                 assert dataset_task_types[adapter_name] == "classification", f"{adapter_name} is not a regression task."
+        self.base_model.to("cuda")
         #print(self.base_model)
     def swith_adapter(self, adapter_name, adapter_id):