Computer Vision , AI

[Torch] requires_grad 와 state_dict() 사이의 관계 본문

備忘錄

[Torch] requires_grad 와 state_dict() 사이의 관계

Elune001 2024. 8. 6. 13:28

state_dict()에는 requires_grad에 대한 정보가 안담기는 것으로 보인다.

 

model.prompt_embeddings.requires_grad

의 return 값은 True여도

model.state_dict()['prompt_embeddings'].requires_grad

의 return 값은 False로 나온다.

 

 

'備忘錄' 카테고리의 다른 글

timm 에러 cannot import name 'container_abcs' from 'torch._six'  (0) 2024.08.07
MMCV 인식 불가 시  (0) 2024.08.07
Deep learning 수학 기초 이론 [Bayes theorem편]  (0) 2022.12.26
StyleGAN  (1) 2022.11.08