cuda out of memory 오류 해결

정또치·2023년 5월 3일

개발일기

목록 보기
11/13

loss 그래프 그리고 싶어서 LOSS 라는 리스트를 만들어 loss 값을 append 했다.

근데 .item() 없이
G_loss.append(loss_generator)
을 해버려서 gpu에 계속 쌓이고 있었다....... 후 .... 하...

그래도 알아서 다행 ~~

profile
ddochi

0개의 댓글