From 325faef9d515c0543c0cf2b746d8ce391120ed1f Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Fri, 31 Jan 2025 17:02:35 +0100 Subject: [PATCH] push --- src/aiunn/finetune.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/aiunn/finetune.py b/src/aiunn/finetune.py index e9c13d6..36ad40a 100644 --- a/src/aiunn/finetune.py +++ b/src/aiunn/finetune.py @@ -375,7 +375,7 @@ if __name__ == "__main__": "/root/training_data/vision-dataset/image_upscaler.parquet", "/root/training_data/vision-dataset/image_vec_upscaler.parquet" ], - batch_size=8, # Increased batch size + batch_size=2, # Increased batch size learning_rate=1e-4 # Reduced initial LR )