Update tasks/base_task.py

tasks/base_task.py

@@ -83,7 +83,7 @@ class BaseTask(nn.Module):
         self.loaded_optimizer_states_dict = {}
         self.trainer = None
         self.logger = None
-        self.on_gpu = True #False
+        self.on_gpu = False
         self.use_dp = False
         self.use_ddp = False
         self.example_input_array = None