diff --git a/configs/train_unrolledADMM.yaml b/configs/train_unrolledADMM.yaml
index 159c85e8..65115aed 100644
--- a/configs/train_unrolledADMM.yaml
+++ b/configs/train_unrolledADMM.yaml
@@ -89,7 +89,7 @@ simulation:
 
 training:
   batch_size: 8
-  epoch: 10
+  epoch: 50
   #In case of instable training
   skip_NAN: True
   slow_start: False  #float how much to reduce lr for first epoch