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)