fix(args): parsing fix

AdityaNG · Mar 3, 2024 · 9b8917c · 9b8917c
1 parent 3e58c93
commit 9b8917c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/LLaVA/llava/train/train.py b/LLaVA/llava/train/train.py
@@ -788,6 +788,9 @@ def make_supervised_data_module(tokenizer: transformers.PreTrainedTokenizer,
                                     data_path=validation_data_path,
                                     data_args=data_args)
         print('Eval dataset found at {}'.format(validation_data_path))
+    else:
+        print('No validation dataset found at {}'.format(validation_data_path))
+        exit(1)
     return dict(train_dataset=train_dataset,
                 eval_dataset=eval_dataset,
                 data_collator=data_collator)

diff --git a/README.md b/README.md
@@ -96,6 +96,7 @@ python3 -m drivellava.scripts.train
 - Training script
   - Select what layers to train
   - [Done] Measure memory requirements: ~ 40 GB vRAM
+  - [Done] [Google Colab training Script](https://colab.research.google.com/drive/1rtZcfZasMIly0xERC9K0gmEXvCIt1vtW#scrollTo=uEsq4SuJZy8C&uniqifier=1)
   - Train on CommaVQ
   - Tabulate results
 - Dataset

diff --git a/drivellava/scripts/train.py b/drivellava/scripts/train.py
@@ -150,7 +150,7 @@ def main():
         --per_device_eval_batch_size 4 \
         --gradient_accumulation_steps 1 \
         --evaluation_strategy "steps" \
-        --eval_steps 50 \
+        --eval_steps 15 \
         --save_strategy "steps" \
         --save_steps 50 \
         --save_total_limit 1 \