Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
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],
|