목록Study/Implementation (1)

soyeonland

model 함수 정리

model.chiildren() - Sequential 구분 여부 model.named_parameters() - 이름과 함께 ('weight','bias') 행렬 텐서만 출력 model.parameters() -이름 없이 ('weight','bias') 행렬 텐서만 출력 model.named_buffers() -pruning 값 저장
Study/Implementation 2020. 3. 24. 14:38
이전 Prev 1 Next 다음