Skip to content

Commit f7df8d6

Browse files
committed
update
1 parent b75b321 commit f7df8d6

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

53 files changed

+12
-12
lines changed
-5.96 KB
Binary file not shown.
-240 Bytes
Binary file not shown.
-7.02 KB
Binary file not shown.
-2.79 KB
Binary file not shown.
-4.75 KB
Binary file not shown.
-2.75 KB
Binary file not shown.

cifar/main.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66
import torch
77
import torch.nn as nn
88
import torch.backends.cudnn as cudnn
9-
import models_bnn
10-
import models
9+
import models_cifar
10+
import models_imagenet
1111
import numpy as np
1212
from torch.autograd import Variable
1313
from utils.options import args
@@ -50,16 +50,16 @@ def main():
5050

5151
if args.dataset=='tinyimagenet':
5252
num_classes=200
53-
model_zoo = 'models.'
53+
model_zoo = 'models_imagenet.'
5454
elif args.dataset=='imagenet':
5555
num_classes=1000
56-
model_zoo = 'models.'
56+
model_zoo = 'models_imagenet.'
5757
elif args.dataset=='cifar10':
5858
num_classes=10
59-
model_zoo = 'models_bnn.'
59+
model_zoo = 'models_cifar.'
6060
elif args.dataset=='cifar100':
6161
num_classes=100
62-
model_zoo = 'models_bnn.'
62+
model_zoo = 'models_cifar.'
6363

6464
if len(args.gpus)==1:
6565
model = eval(model_zoo+args.model)(num_classes=num_classes).cuda()
-152 Bytes
Binary file not shown.
-4.64 KB
Binary file not shown.
-292 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)