Pytorch 模型的網絡結構可視化

Keras 中 keras.summary() 即可很好的將模型結構可視化，但 Pytorch 暫還沒有提供網絡模型可視化的工具. 總結兩種pytorch網絡結構的可視化方法 Pytorch使用Tensorboard可視化網絡結構 GitHub地址：點擊打開 1.下載可視化代碼 git clone https://github.com/lanpa/tensorboard-pytorch.git 2

>>阅读原文<<