pytorch-beginner
pytorch-beginner copied to clipboard
04-Convolutional-Neural-Network两个问题
1from logger import Logger有警告 2D:\Anaconda3\python.exe G:/pytorch/pytorch-beginner-master/04-Convolutional-Neural-Network/convolution_network.py epoch 1
Traceback (most recent call last):
File "G:/pytorch/pytorch-beginner-master/04-Convolutional-Neural-Network/convolution_network.py", line 78, in
Process finished with exit code 1
出现:“IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
”,这是PyTorchd 版本问题。
将.data[0]全部改为.item()。
希望可以解决您的问题,祝好!
好的,谢谢