首頁 > 軟體

使用PyTorch常見4個錯誤解決範例詳解

2022-10-16 14:01:39

導讀

這4個錯誤,我敢說大部分人都犯過,希望能給大家一點提醒。

最常見的神經網路錯誤:

1)你沒有首先嚐試過擬合單個batch。

2)你忘了為網路設定train/eval模式。

3)在.backward()之前忘記了.zero_grad()(在pytorch中)。

4)將softmaxed輸出傳遞給了期望原始logits的損失,還有其他嗎?


IT145.com E-mail:sddin#qq.com