공부하는 코드가 pytorch 1.2.0 버전인지라 너무 구버전인 것 같아 pytorh 2.1.2로 가상환경을 세팅하였다.
와중에 아래 코드에서
이 사람처럼 각 loss에 대해 따로 zero_grad() -> backward() -> step()을 하지 않고 아래 순서로 코드를 바꿔주니 잘 돌아간다.
[Solved][Pytorch1.5] RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operatio
Hi, This is most likely happening because value_optimizer.step() actually modifies the weights of the model inplace while the original value of these weights is needed to compute action_loss.backward(). Is that the issue?
discuss.pytorch.org
'고군분투' 카테고리의 다른 글
| Dataloader (TypeError: cannot pickle 'module' object) (0) | 2024.05.14 |
|---|---|
| cmd 창 안 열리는 현상 해결 (0) | 2022.09.02 |