From 869330f7c51921368c07ff15184a7e916f121346 Mon Sep 17 00:00:00 2001
From: tremblerz <abhishek.s14@iiits.in>
Date: Sat, 17 Aug 2024 02:58:49 +0000
Subject: [PATCH] keep a small model as default

---
 src/configs/algo_config.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/configs/algo_config.py b/src/configs/algo_config.py
index 1a9ff55..d98c5de 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": [],