저처럼 미친 짓하시면 안 됩니다....
잘못된 예시
model.requires_grad = False
문제는 위 코드를 넣고 돌려도 에러가 발생하지 않아 GPU 사용량이 달라지거나 파라미터를 따로 확인하지 않는 한,
제대로 된 예시
for param in model.parameters():
param.requires_grad = False
어떤 블로그에서 보고 그대로 긁어서 했었는데 그 블로그가 잘못된 것 같아요. 물론 지금 검색해보니 좋은 자료가 많이 나오네요. 제 불찰입니다. 여하튼 반복하지 않기를 바라며 강조해봅니다.
'NLP > error' 카테고리의 다른 글
도커 컨테이너 유지한 채 마운트 경로 추가하기 (0) | 2024.09.03 |
---|---|
[error] conda 에서 local 에 설치된 라이브러리에 접근해요 (0) | 2024.02.23 |
[error] GPU가 충분한데 CUDA OUT OF MEMORY가 발생합니다 (0) | 2023.12.13 |
[error] PeftModelForCausalLM.generate() takes 1 positional argument and 2 were given (0) | 2023.09.19 |
[error] 파이썬 엑셀 인코딩 문제 해결 (0) | 2023.09.19 |