From 3a34bf8b74a3adcc4157ac88e0c54a83a676b9c4 Mon Sep 17 00:00:00 2001 From: IshanRattan Date: Sat, 6 Apr 2024 15:27:17 +0530 Subject: [PATCH] updated train.py --- dl/image/classification/cifar10/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/dl/image/classification/cifar10/train.py b/dl/image/classification/cifar10/train.py index 3d522d2..595fe04 100644 --- a/dl/image/classification/cifar10/train.py +++ b/dl/image/classification/cifar10/train.py @@ -1,6 +1,7 @@ from image_transformation import transformations +import torch.nn as nn import torchvision train_set = torchvision.datasets.CIFAR10(root=data_path, train=True, transform=transformations()['train'],