Pytorch Tips

Pytorch 模型的网络结构可视化

参考: https://blog.csdn.net/TTdreamloong/article/details/83107110

可能遇到的报错:

• AttributeError: 'torch._C.Value' object has no attribute 'debugName'
  原因: torch版本有误

  # 列举pip当前可以更新的所有安装包
  pip list --outdated
  # 使用Pip更新Pytorch和torchvision
  pip install --upgrade pytorch torchvision
  

• 提示安装"Graphviz"
  原因: 需要在系统环境下安装,并非虚拟环境

  sudo apt-get install graphviz