pytorch 坑

文章目录

    • 1、`AttributeError: 'module' object has no attribute 'Resize'?`
    • 2. nn.Flatten()


1、AttributeError: 'module' object has no attribute 'Resize'?

原因:版本问题
解决办法: conda uninstall torchvision

         pip install torchvision --trusted-host pypi.mirrors.ustc.edu.cn`

2. nn.Flatten()

原因:提示无该模块,但其实torch/nn/modules文件夹下有flatten.py,只是没加载出来
解决办法:
1.进入nn模块(pycharm,按住Ctrl,点击nn)
pytorch 坑_第1张图片
2.进入.modules 模块(pycharm,按住Ctrl,点击.modules)
pytorch 坑_第2张图片
3.在进入.modules 后,最后一行加上

from .flatten import Flatten as Flatten

你可能感兴趣的:(pytorch,pytorch,深度学习)