RuntimeError: DataLoader worker (pid(s) 9528, 8320) exited unexpectedly問題解決記錄

僅供學習記錄,若有侵權可聯繫刪除。 方法一: num_workers設置爲0,默認值爲0表示不啓用多進程。該方法在處理大量數據的時候還是不行,肯定要多進程。參見方法二。 方法二:將涉及多線程的代碼放到if name == 'main’的範圍內 if name == ‘main’: for epoch in range(3): for step, (batch_x, batch_y) in enum
相關文章
相關標籤/搜索