diff --git a/config.json b/config.json
index 9ca3462..906bf2d 100644
--- a/config.json
+++ b/config.json
@@ -29,6 +29,7 @@
   "image_size": 224,
   "initializer_range": 0.02,
   "layer_norm_eps": 1e-06,
+  "max_depth": 10,
   "mlp_ratios": [
     4,
     4,