Skip to content

Commit e7f5457

Browse files
authored
async --> non_blocking
1 parent 2395385 commit e7f5457

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

AutoML/darts-master/cnn/train_imagenet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -209,7 +209,7 @@ def infer(valid_queue, model, criterion):
209209

210210
for step, (input, target) in enumerate(valid_queue):
211211
input = Variable(input, volatile=True).cuda()
212-
target = Variable(target, volatile=True).cuda(async=True)
212+
target = Variable(target, volatile=True).cuda(non_blocking=True)
213213

214214
logits, _ = model(input)
215215
loss = criterion(logits, target)

0 commit comments

Comments
 (0)