undo

YuanTingHsieh · YuanTingHsieh · commit 2ca2acecd178 · 2025-02-06T21:05:22.000-08:00
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")
