some modification

Coobiw · Oct 24, 2023 · 804d9b3 · 804d9b3
1 parent 2e2eb77
commit 804d9b3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/cli_demo.py b/cli_demo.py
@@ -153,6 +153,9 @@ def main():
     first = True
     while True:
         query = _get_input()
+
+        if not history:
+            first = True
 
         # Process commands.
         if query.startswith(':'):

diff --git a/lavis/projects/instruction_tuning/train.yaml b/lavis/projects/instruction_tuning/train.yaml
@@ -76,7 +76,7 @@ run:
   warmup_steps: 500
   weight_decay: 0.05
   grad_norm_clip: 1.
-  max_epoch: 5 #5
+  max_epoch: 10 #5
   batch_size_train: 1 #16
   batch_size_eval: 1
   num_workers: 4
@@ -87,7 +87,7 @@ run:
   num_beams: 5 # 同上
 
   seed: 42
-  output_dir: "output/instruction_tuning"
+  output_dir: "output/instruction_tuning/lr1e-4"
 
   amp: False # True
   resume_ckpt_path: null