help wantedvision
Repository metrics
- Stars
- (21,634 stars)
- PR merge metrics
- (30d に merged 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'