codetag-kr.dev
git
java
c++
c#
c
r
.net
string
android
git-branch
Dark
neural network
[1/1]
PyTorch에서 register_parameter와 register_buffer의 차이점
1. 최적화register_parameter로 추가된 속성은 모델 학습 과정에서 자동으로 최적화됩니다.register_buffer로 추가된 속성은 최적화에 참여하지 않습니다.2. 데이터 유형register_parameter는 학습 가능한 변수 (Tensor)만 추가할 수 있습니다
1