From 6aad1a38f5bc0435bef752a627d6624d87583d72 Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Tue, 25 Feb 2025 15:47:50 +0100 Subject: [PATCH] increased trainingset amount --- example.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example.py b/example.py index 8492d51..b2c26c8 100644 --- a/example.py +++ b/example.py @@ -98,7 +98,7 @@ if __name__ =="__main__": "/root/training_data/vision-dataset/image_vec_upscaler.parquet" ], 'transform': transforms.Compose([transforms.ToTensor()]), - 'samples_per_file': 2500 + 'samples_per_file': 5000 } trainer.load_data(dataset_params=dataset_params, batch_size=1)