We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 7e5d8e1 + 3d829de commit 3d02706Copy full SHA for 3d02706
benchmark_models.py
@@ -37,7 +37,7 @@ def train(type='single'):
37
"""use fake image for training speed test"""
38
img = Variable(torch.randn(args.BATCH_SIZE, 3, 224, 224)).cuda()
39
target = Variable(torch.LongTensor(args.BATCH_SIZE).random_(args.NUM_CLASSES)).cuda()
40
- criterion = nn.CrossEntropyLoss()dock
+ criterion = nn.CrossEntropyLoss()
41
benchmark = {}
42
for model_type in MODEL_LIST.keys():
43
for model_name in MODEL_LIST[model_type]:
0 commit comments