From d66b167b857542130b8bc46f660f42f9db3b1c58 Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Fri, 21 Feb 2025 21:17:40 +0100 Subject: [PATCH] downsized abtchsize to 8 --- 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 b47478c..b1ee2e5 100644 --- a/src/aiunn/finetune.py +++ b/src/aiunn/finetune.py @@ -94,7 +94,7 @@ from torch.utils.data import DataLoader # Create your dataset and dataloader dataset = UpscaleDataset(["/root/training_data/vision-dataset/image_upscaler.parquet", "/root/training_data/vision-dataset/image_vec_upscaler.parquet"], transform=transform) -data_loader = DataLoader(dataset, batch_size=16, shuffle=True) +data_loader = DataLoader(dataset, batch_size=8, shuffle=True) # Define a loss function and optimizer criterion = nn.MSELoss()