updated train.py

IshanRattan · Aug 7, 2024 · 3a34bf8 · 3a34bf8
1 parent 215e68b
commit 3a34bf8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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'],