pytorch volatile

发表于 | 阅读次数: 
1
2
3
4
5
6
if args.cuda:
    images_test = Variable(images_test.cuda(), volatile=True)
    targets_test = [Variable(ann.cuda(), volatile=True) for ann in targets_test]
else:
    images_test = Variable(images_test, volatile=True)
    targets_test = [Variable(ann, volatile=True) for ann in targets_test]
请作者喝一杯咖啡☕️