728x90
다음 코드에서 numpy.ndarray를 torch.tensor로 변환하는 중 에러가 발생하였다.
last_layer.weight.data = torch.tensor(last).unsqueeze(-1).unsqueeze(-1)
다음과 같이 torch.from_numpy()와 copy()를 사용하여 해결하였다.
last_layer.weight.data = torch.from_numpy(last.copy()).unsqueeze(-1).unsqueeze(-1)
[ref]
728x90
'Temp' 카테고리의 다른 글
[Pillow] 이미지를 읽었더니 회전될 때 (0) | 2021.12.13 |
---|---|
[Ubuntu] SSH 파일을 사용하여 Local -> Remote(Server) 파일 전송 (0) | 2021.12.01 |
[Filezilla] SSH 파일을 사용하여 Local -> Remote(Server) 파일 전송 (0) | 2021.11.30 |
[Colab] 연결 끊김 방지 (0) | 2021.11.29 |
[Pytorch] Optimizer learning rate 가져오기 (0) | 2021.11.27 |