윤제로의 제로베이스

AssertionError: if capturable=false, state_steps should not be cuda tensors. 본문

Issue board

AssertionError: if capturable=false, state_steps should not be cuda tensors.

윤_제로 2023. 5. 25. 02:25

NeRF pytorch 코드를 돌리다가 자꾸 이 에러가 뜨길래 에러 해결 법을 찾아보았다.

 

optimizer = optim.Adam(...)
optimizer.param_groups[0]['capturable'] = True

optimizer 에서 'capturable'을 true로 바꿔줘야한다...

 

https://normal-engineer.tistory.com/321

 

[에러기록] assertionerror: if capturable=false, state_steps should not be cuda tensors.

pytorch 안에서 adam을 쓸 때 자꾸 위와 같은 에러가 떠서 이를 위해 optimizer = optim.Adam(func.parameters(), lr=args.lr) optimizer.param_groups[0]['capturable'] = True 다음과 같이 'capturable'을 true로 바꿔줘야했다. 그런

normal-engineer.tistory.com