Update number epochs.

LCAV · Sep 14, 2023 · 60f1f5c · 60f1f5c
1 parent 1dc5a32
commit 60f1f5c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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