Closed ssaru closed 5 years ago
model.to("cuda:0")로 모델이 GPU로 복사되면 에러가 납니다. 아마도 binarized weight가 model의 Parameter로 등록되지 않아서 binarized weight는 gpu로 복사가 안되서 발생하는 문제 같습니다.
model.to("cuda:0")
해당 문제 한번 살펴보겠습니다.
네 알겠습니다.
Ronalmoo
commented
5 years ago Ronalmoo
commented
5 years ago
model.to("cuda:0")로 모델이 GPU로 복사되면 에러가 납니다. 아마도 binarized weight가 model의 Parameter로 등록되지 않아서 binarized weight는 gpu로 복사가 안되서 발생하는 문제 같습니다.해당 문제 한번 살펴보겠습니다.