search

chenyuntc / pytorch-book

PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch：入门与实战》)
MIT License
11.99k stars 3.73k forks source link

第十章 無法訓練模型 #191

Open AlanHuang1998 opened 4 years ago

AlanHuang1998 commented 4 years ago

照著github上的方法做了一遍，demo.ipynb可以正常執行，但想要訓練自己的模型時，執行python main.py train就會遇到下面這個問題，問題看起來是出現main.py在88行的迴圈，是不是有什麼辦法能解決? 1572676423021

註: pytorch = 1.3, python = 3.7