undo

NVIDIA · Feb 7, 2025 · 2ca2ace · 2ca2ace
1 parent 6ef8464
commit 2ca2ace
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/hello-world/hello-pt/src/hello-pt_cifar10_fl.py b/examples/hello-world/hello-pt/src/hello-pt_cifar10_fl.py
@@ -30,7 +30,7 @@
 
 def main():
     batch_size = 4
-    epochs = 2
+    epochs = 5
     lr = 0.01
     model = SimpleNetwork()
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")