From 167aceac2afa66a4d459f984cd245d9d25f197a1 Mon Sep 17 00:00:00 2001 From: joaqo Date: Sat, 8 Apr 2017 16:26:51 -0300 Subject: [PATCH] Fix typo in imagenet/main.py --- imagenet/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/imagenet/main.py b/imagenet/main.py index 86abea3739..f0f196a1e5 100644 --- a/imagenet/main.py +++ b/imagenet/main.py @@ -112,7 +112,7 @@ def main(): batch_size=args.batch_size, shuffle=False, num_workers=args.workers, pin_memory=True) - # define loss function (criterion) and pptimizer + # define loss function (criterion) and optimizer criterion = nn.CrossEntropyLoss().cuda() optimizer = torch.optim.SGD(model.parameters(), args.lr,