pytorch/examples

InceptionV3 cannot work!

Open

#362 aberto em 27 de mai. de 2018

Ver no GitHub
 (8 comments) (0 reactions) (0 assignees)Python (9.429 forks)batch import
help wantedvision

Métricas do repositório

Stars
 (21.634 stars)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

Description

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'

Guia do colaborador

Direção de pesquisa
Verifique a versão do PyTorch; o argumento non blocking para .cuda() foi introduzido no PyTorch 0.4.0. Atualize o PyTorch para >=0.4.0 ou remova o argumento non blocking=True da chamada target.cuda().
Pilha de tecnologia
pythonpytorch
Domain
machine learning
Tipo Issue
Bug
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
2
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
Menos de 1 hora
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
GitPython
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
75