Skip to content

Commit 3d02706

Browse files
authored
Merge pull request #2 from johmathe/master
Update typo
2 parents 7e5d8e1 + 3d829de commit 3d02706

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

benchmark_models.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def train(type='single'):
3737
"""use fake image for training speed test"""
3838
img = Variable(torch.randn(args.BATCH_SIZE, 3, 224, 224)).cuda()
3939
target = Variable(torch.LongTensor(args.BATCH_SIZE).random_(args.NUM_CLASSES)).cuda()
40-
criterion = nn.CrossEntropyLoss()dock
40+
criterion = nn.CrossEntropyLoss()
4141
benchmark = {}
4242
for model_type in MODEL_LIST.keys():
4343
for model_name in MODEL_LIST[model_type]:

0 commit comments

Comments
 (0)