pytorch-book
pytorch-book copied to clipboard
第十章 無法訓練模型
照著github上的方法做了一遍,demo.ipynb可以正常執行,但想要訓練自己的模型時,執行python main.py train就會遇到下面這個問題,問題看起來是出現main.py在88行的迴圈,是不是有什麼辦法能解決?
註: pytorch = 1.3, python = 3.7