help wantedvision
仓库指标
- Star
- (21,634 star)
- PR 合并指标
- (30 天内没有已合并 PR)
描述
python main.py -a inception_v3 ./imagenet/cat2dog --batch-size 16 --print-freq 1 --pretrained;
=> using pre-trained model 'inception_v3'
Traceback (most recent call last):
File "main.py", line 314, in
main()
File "main.py", line 157, in main
train(train_loader, model, criterion, optimizer, epoch)
File "main.py", line 189, in train
target = target.cuda(non_blocking=True)
TypeError: _cuda() got an unexpected keyword argument 'non_blocking'