혹시 loss값을 기록하려는 목적에서 loss_list.append(loss)같은 코드를 쓰셨나요? 그러신 경우 loss 대신 loss.item()을 append하면 문제가 해결될 것으로 예상합니다. loss를 지우지 않고 쌓아두는 경우 computation graph가 점점 쌓여서 backward가 느려집니다.
'Debug(디버그)' 카테고리의 다른 글
plt.imshow 함수 실행시 the kernel appears to have died. it will restart automatically. 뜨는 경우 (0) | 2022.08.09 |
---|