목차 728x90 torch의 버전 문제로 0.5 미만 버전은 loss 추출할 때 loss[0].item() 위의 코드를 사용했었다. loss.item() 이 코드로 수정하면 된다. 728x90 공유하기 게시글 관리 뚝딱이 'ERROR' 카테고리의 다른 글 [Selenium] NoSuchDriverException: Message: Unable to obtain chromedriver using Selenium Manager; 'str' object has no attribute 'capabilities'; For documentation on this error, please visit: https://www.selenium.dev/documentation/webdriver/troubleshootin.. (0) 2023.07.03 [heapq] 파이썬 최소힙 pop()에서 최솟값이 안나올 때 (0) 2023.05.02 [DL] RuntimeError: DataLoader worker (pid(s) 106594, 106595) exited unexpectedly (0) 2023.04.25 [DL] RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start method (0) 2023.04.25 [DL] RuntimeError: Expected a 'cuda' device type for generator but found 'cpu' (0) 2023.04.25 목차 목차 닫기