From 3ab3de4e640e12cf03b31c3585b97f4f39295d6a Mon Sep 17 00:00:00 2001
From: Yu Li <lyo.gavin@gmail.com>
Date: Fri, 15 Sep 2023 22:07:57 -0500
Subject: [PATCH] fix script for training

---
 anima_100k/run_longer_training.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/anima_100k/run_longer_training.sh b/anima_100k/run_longer_training.sh
index 7eea03d..e6b8efa 100755
--- a/anima_100k/run_longer_training.sh
+++ b/anima_100k/run_longer_training.sh
@@ -18,7 +18,7 @@ mkdir -p $OUTPUT_PATH
 
 
 python longer_training.py --dataset="DATASET_PATH" \
-    --dataset_format="FORMAT" \
+    --dataset_format="long_data" \
     --learning_rate 0.0001 \
     --per_device_train_batch_size 1 \
     --gradient_accumulation_steps 16 \