update examples

Lightning-AI · Feb 16, 2023 · 747a08c · 747a08c
1 parent af09b84
commit 747a08c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/app_multi_node/train_fabric.py b/examples/app_multi_node/train_fabric.py
@@ -15,7 +15,7 @@ def run(self):
         )
 
         # 2. Create Fabric.
-        fabric = Fabric(strategy="ddp", precision=16)
+        fabric = Fabric(strategy="ddp", precision='16-mixed')
         model, optimizer = fabric.setup(model, torch.optim.SGD(model.parameters(), lr=0.01))
         criterion = torch.nn.MSELoss()
 

diff --git a/examples/pl_hpu/mnist_sample.py b/examples/pl_hpu/mnist_sample.py
@@ -63,7 +63,7 @@ def configure_optimizers(self):
             "accelerator": "hpu",
             "devices": 1,
             "max_epochs": 1,
-            "plugins": lazy_instance(HPUPrecisionPlugin, precision=16),
+            "plugins": lazy_instance(HPUPrecisionPlugin, precision='16-mixed'),
         },
         run=False,
         save_config_kwargs={"overwrite": True},