convlution = CONV_net(input_channels = 3, output_size = 10)
net_work = create_network(convlution)
image_size = 32
transform = resize_image(image_size)
BATCH_SIZE = 64
def load_data(transform, BATCH_SIZE, train, PATH = ‘Data’):
load_dataset = datasets.CIFAR100(PATH,
train=train,
transform=transform,
download=False)
data_loader = torch.utils.data.DataLoader(dataset = load_dataset, 
                                        batch_size = BATCH_SIZE, shuffle = True)

return data_loader

train_loader=load_data(transform, BATCH_SIZE, train= True, PATH = ‘Data’)
test_loader=load_data(transform, BATCH_SIZE, train = False, PATH = ‘Data’)
EPOCH = 1
LR = 0.001
trained_model = train_model(net_work, train_loader, LR, EPOCH)
this is my code below it, so what should I do to fix the erroe?
[/quote]