Skip to content

S
SCL-my
Commit ab636756 by yuxguo
Options

fix

parent 18237e29
Showing with 3 additions and 3 deletions
main.py
......@@ -370,7 +370,7 @@ def train(trainloader, optimizer, model, epoch):
counter = 0
for batch_idx, (images, label, meta_target) in enumerate(trainloader):
counter += 1
if args.use_cuda:
if args.use_gpu:
images = images.cuda()
label = label.cuda()
meta_target = meta_target.cuda()
......@@ -401,7 +401,7 @@ def validate(validloader, model, epoch):
with torch.no_grad():
for batch_idx, (images, label, meta_target) in enumerate(validloader):
counter += 1
if args.use_cuda:
if args.use_gpu:
images = images.cuda()
label = label.cuda()
meta_target = meta_target.cuda()
......@@ -427,7 +427,7 @@ def test(testloader, model, epoch):
with torch.no_grad():
for batch_idx, (images, label, meta_target) in enumerate(testloader):
counter += 1
if args.use_cuda:
if args.use_gpu:
images = images.cuda()
label = label.cuda()
meta_target = meta_target.cuda()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment