Abdullah-Nazhat
/

Normalizer

Model card Files Files and versions

Abdullah-Nazhat commited on Jul 1

Commit

7fb376d

·

verified ·

1 Parent(s): f9f0532

Update train.py

Files changed (1) hide show

train.py +2 -3

train.py CHANGED Viewed

@@ -10,10 +10,9 @@ from normalizer import Normalizer
 transform = Compose([
-RandomCrop(32, padding=4),
-RandomHorizontalFlip(),
 ToTensor(),
-Normalize((0.5, 0.5,0.5),(0.5, 0.5,0.5))
 ])

 transform = Compose([
 ToTensor(),
+Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
 ])