From 153ead15b7cc7e33bdc4a714c2f71e10f4390036 Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Mon, 24 Feb 2025 16:32:23 +0100 Subject: [PATCH] downsized image amount --- 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 03efb8f..0013ec9 100644 --- a/src/aiunn/finetune.py +++ b/src/aiunn/finetune.py @@ -41,7 +41,7 @@ class UpscaleDataset(Dataset): combined_df = pd.DataFrame() for parquet_file in parquet_files: # Load a subset (head(2500)) from each parquet file - df = pd.read_parquet(parquet_file, columns=['image_512', 'image_1024']).head(2500) + df = pd.read_parquet(parquet_file, columns=['image_512', 'image_1024']).head(1250) combined_df = pd.concat([combined_df, df], ignore_index=True) # Validate rows (ensuring each value is bytes or str)