From f701c4e34bccc46cc8cf4050c1d1bb4bcf3e5385 Mon Sep 17 00:00:00 2001 From: SamimAB Date: Thu, 31 Aug 2023 03:21:35 +0530 Subject: [PATCH] Changes to be committed: modified: train_configs/minigpt4_llama2_stage2_finetune.yaml --- train_configs/minigpt4_llama2_stage2_finetune.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_configs/minigpt4_llama2_stage2_finetune.yaml b/train_configs/minigpt4_llama2_stage2_finetune.yaml index 91b4b7d..052c1ae 100644 --- a/train_configs/minigpt4_llama2_stage2_finetune.yaml +++ b/train_configs/minigpt4_llama2_stage2_finetune.yaml @@ -28,7 +28,7 @@ run: warmup_lr: 1e-6 weight_decay: 0.05 - max_epoch: 5 + max_epoch: 100 iters_per_epoch: 200 batch_size_train: 12 batch_size_eval: 12