@@ -7,7 +7,7 @@ python train.py \
-bs 16 \
-size 640 \
--wp_epoch 1 \
- --max_epoch 150 \
+ --max_epoch 250 \
--eval_epoch 10 \
--ema \
--fp16 \