From bd351734064190a37577cfcb6687b208bac1ee8d Mon Sep 17 00:00:00 2001 From: thomas chaton Date: Thu, 10 Nov 2022 12:42:19 +0000 Subject: [PATCH] [App] Resolve multi-node cloud bug (#15619) update (cherry picked from commit 47314eafbd62fd9abd7b73c9edbb91065ee32b07) --- examples/app_multi_node/train_lite.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/app_multi_node/train_lite.py b/examples/app_multi_node/train_lite.py index ed9777a1064f6..feb8ac2226b77 100644 --- a/examples/app_multi_node/train_lite.py +++ b/examples/app_multi_node/train_lite.py @@ -9,7 +9,7 @@ class LitePyTorchDistributed(L.LightningWork): @staticmethod def run(): # 1. Create LightningLite. - lite = LightningLite(strategy="ddp", precision="bf16") + lite = LightningLite(strategy="ddp", precision=16) # 2. Prepare distributed model and optimizer. model = torch.nn.Linear(32, 2)