From 804882d7a2a85535776db4cfbd629530fb44b962 Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Fri, 31 Jan 2025 16:11:21 +0100 Subject: [PATCH] corrected dataset paths --- src/aiunn/finetune.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/aiunn/finetune.py b/src/aiunn/finetune.py index 3e7687a..1fefbe4 100644 --- a/src/aiunn/finetune.py +++ b/src/aiunn/finetune.py @@ -397,8 +397,8 @@ if __name__ == "__main__": trainer = FineTuner( model=model, dataset_paths=[ - "/root/training_data/vision-dataset/image_upscaler.0", - "/root/training_data/vision-dataset/image_vec_upscaler.0" + "/root/training_data/vision-dataset/image_upscaler.parquet", + "/root/training_data/vision-dataset/image_vec_upscaler.parquet" ], batch_size=8, # Increased batch size learning_rate=1e-4 # Reduced initial LR