diff --git a/train_configs/minigptv2_finetune_mvtec.yaml b/train_configs/minigptv2_finetune_mvtec.yaml
index 6cfd653..3de8714 100644
--- a/train_configs/minigptv2_finetune_mvtec.yaml
+++ b/train_configs/minigptv2_finetune_mvtec.yaml
@@ -33,7 +33,7 @@ run:
   warmup_lr: 1e-6
 
   weight_decay: 0.05
-  max_epoch: 5
+  max_epoch: 2
   num_workers: 6
   warmup_steps: 1000
   iters_per_epoch: 2000
diff --git a/train_configs/minigptv2_finetune_textvqa.yaml b/train_configs/minigptv2_finetune_textvqa.yaml
index 9755295..b7160a8 100644
--- a/train_configs/minigptv2_finetune_textvqa.yaml
+++ b/train_configs/minigptv2_finetune_textvqa.yaml
@@ -33,7 +33,7 @@ run:
   warmup_lr: 1e-6
 
   weight_decay: 0.05
-  max_epoch: 5
+  max_epoch: 2
   num_workers: 6
   warmup_steps: 1000
   iters_per_epoch: 2000