keep a small model as default

aidecentralized · Aug 17, 2024 · 869330f · 869330f
1 parent 43d016e
commit 869330f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/configs/algo_config.py b/src/configs/algo_config.py
@@ -22,7 +22,7 @@
     "exp_type": "iid_clients_federated",
     # Learning setup
     "epochs": 1000,
-    "model": "resnet34",
+    "model": "resnet10",
     "model_lr": 3e-4,
     "batch_size": 256,
     "exp_keys": [],