LN1996 commited on
Commit
db44e50
1 Parent(s): 2cdff8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ model.eval()
19
  classes = ('plane', 'car', 'bird', 'cat', 'deer',
20
  'dog', 'frog', 'horse', 'ship', 'truck')
21
 
22
- cifar_valid = utils.Cifar10SearchDataset('.', train=False, download=False, transform=utils.augmentation_custom_resnet('Valid'))
23
 
24
  inv_normalize = transforms.Normalize(
25
  mean=[-0.50/0.23, -0.50/0.23, -0.50/0.23],
 
19
  classes = ('plane', 'car', 'bird', 'cat', 'deer',
20
  'dog', 'frog', 'horse', 'ship', 'truck')
21
 
22
+ cifar_valid = utils.Cifar10SearchDataset('.', train=False, download=True, transform=utils.augmentation_custom_resnet('Valid'))
23
 
24
  inv_normalize = transforms.Normalize(
25
  mean=[-0.50/0.23, -0.50/0.23, -0.50/0.23],