From da73e223e3caba033eca63df8203107a00e71393 Mon Sep 17 00:00:00 2001 From: s473558 Date: Mon, 5 Jun 2023 04:48:04 +0200 Subject: [PATCH] add more code flexibility --- neural_network/datasets.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/neural_network/datasets.py b/neural_network/datasets.py index 33c7f162..166268c5 100644 --- a/neural_network/datasets.py +++ b/neural_network/datasets.py @@ -29,9 +29,9 @@ valid_transform = transforms.Compose([ ) ]) -train_dataset = torchvision.datasets.ImageFolder(root='./Vegetable Images/train', transform=train_transform) +train_dataset = torchvision.datasets.ImageFolder(root='./images/train', transform=train_transform) -validation_dataset = torchvision.datasets.ImageFolder(root='./Vegetable Images/validation', transform=valid_transform) +validation_dataset = torchvision.datasets.ImageFolder(root='./images/validation', transform=valid_transform) train_loader = DataLoader( train_dataset, batch_size=BATCH_SIZE, shuffle=True, num_workers=0, pin_memory=True