pytorch實戰遇到的bugs

在訓練和測試的時候都遇到這個問題，

_error_if_any_worker_fails()

RuntimeError: DataLoader worker (pid 8565) is killed by signal: Killed.

 

還有這個，

RuntimeError: $ Torch: not enough memory: you tried to allocate

 

經過研究，修改batch_size參數就不報錯了。將batch_size改小一點，甚至是1都可以。