KeyError: Caught KeyError in replica 0 on device 0.

跑torch代碼遇到這個問題。
博客上有類似的問題：
運行開源庫CCPD-RPnet代碼，提示「KeyError: Caught KeyError in replica 0 on device 0」錯誤

不過具體的錯誤提示和我的還是不一樣。
通過官方論壇裏的一個問題

主要問題是出現在DataParrallel上。
當在多個gpu上跑代碼的時候就可能遇到這個問題。
解決方法就是用一塊GPU跑。