PyTorch Paper Replicating

[AhmedAlziadi]

Hello Great Daniel .

Im trying to run this piece of code but this is the error i got . I dont know what I should do!

the code 👍

Set the batch size

BATCH_SIZE = 32 # this is lower than the ViT paper but it's because we're starting small

Create data loaders

train_dataloader, test_dataloader, class_names = data_setup.create_dataloaders(
train_dir=train_dir,
test_dir=test_dir,
transform=manual_transforms, # use manually created transforms
batch_size=BATCH_SIZE
)

train_dataloader, test_dataloader, class_names

the output 👎

NameError Traceback (most recent call last)
in <cell line: 5>()
4 # Create data loaders
5 train_dataloader, test_dataloader, class_names = data_setup.create_dataloaders(
----> 6 train_dir=train_dir,
7 test_dir=test_dir,
8 transform=manual_transforms, # use manually created transforms

NameError: name 'train_dir' is not defined