molecular_property_prediction_zero_gpu

Running on Zero

feiyang-cai commited on Jan 23

Commit

423e3eb

verified ·

1 Parent(s): 73aeadd

print device

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -230,7 +230,7 @@ class MolecularPropertyPredictionModel():
                 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):
         # return flag:

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