todo

Lightning-AI · Sep 28, 2022 · 6fff52d · 6fff52d
1 parent cc2ea3d
commit 6fff52d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lightning_lite/lite.py b/src/lightning_lite/lite.py
@@ -161,7 +161,7 @@ def setup(
 
         # Let accelerator/plugin wrap and connect the models and optimizers
         model, optimizers = self._strategy.setup_module_and_optimizers(model, list(optimizers))
-        optimizers_to_device(optimizers, self.device)
+        optimizers_to_device(optimizers, self.device)  # TODO: debugging, seems to make no difference
         model = _LiteModule(model, self._precision_plugin, original_module=original_model)
 
         # Update the _DeviceDtypeModuleMixin's device parameter