모델 파라미터 개수 확인하기

황준하·2024년 7월 6일

require_grad 옵션이 True인 parameter 개수 확인

sum(p.numel() for p in model.parameters() if p.requires_grad)



0개의 댓글