diff --git a/configs/gaitedge/phase2_gaitedge.yaml b/configs/gaitedge/phase2_gaitedge.yaml
index 5a70e50..32979e1 100644
--- a/configs/gaitedge/phase2_gaitedge.yaml
+++ b/configs/gaitedge/phase2_gaitedge.yaml
@@ -51,6 +51,7 @@ model_cfg:
   edge: true
   align: true
   seg_lr: 0.00001
+  kernel_size: 3
 
 optimizer_cfg:
   lr: 0.0001