diff --git a/src/configs/algo_config.py b/src/configs/algo_config.py
index 9f1044e..95ff516 100644
--- 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": [],