There was an error while loading. Please reload this page.
1 parent 50afe29 commit 28471f2Copy full SHA for 28471f2
imagenet/main.py
@@ -112,7 +112,7 @@ def main():
112
batch_size=args.batch_size, shuffle=False,
113
num_workers=args.workers, pin_memory=True)
114
115
- # define loss function (criterion) and pptimizer
+ # define loss function (criterion) and optimizer
116
criterion = nn.CrossEntropyLoss().cuda()
117
118
optimizer = torch.optim.SGD(model.parameters(), args.lr,
0 commit comments