From 7bb8ca2a388a1d48024d149961e1c24e43b5f324 Mon Sep 17 00:00:00 2001 From: ThuanNaN Date: Mon, 20 Jan 2025 07:29:12 +0700 Subject: [PATCH] update max epoch to 2 --- train_configs/minigptv2_finetune_mvtec.yaml | 2 +- train_configs/minigptv2_finetune_textvqa.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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