hwang_junha
로그인
hwang_junha
로그인
모델 파라미터 개수 확인하기
황준하
·
2024년 7월 6일
팔로우
0
PyTorch
require_grad 옵션이 True인 parameter 개수 확인
sum
(
p
.
numel
(
)
for
p
in
model
.
parameters
(
)
if
p
.
requires_grad
)
Ref)
https://discuss.pytorch.org/t/how-do-i-check-the-number-of-parameters-of-a-model/4325/8
황준하
팔로우
이전 포스트
[huggingface] not enough values to unpack (expected 2, got 1)
다음 포스트
safetensor 모델 파일 load
0개의 댓글
댓글 작성