image_caption/train.py - Githubissues

yunjey / pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers

MIT License

29.79k stars 8.03k forks source link

Closed CherishineNi closed 5 years ago

CherishineNi commented 5 years ago

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

AttributeError: 'module' object has no attribute 'device'