1.5 KiB
1.5 KiB
사용자 정의 레이어 및 유틸리티 custom-layers-and-utilities
이 페이지는 라이브러리에서 사용되는 사용자 정의 레이어와 모델링을 위한 유틸리티 함수들을 나열합니다.
이 함수들 대부분은 라이브러리 내의 모델 코드를 연구할 때만 유용합니다.
PyTorch 사용자 정의 모듈 transformers.Conv1D
autodoc pytorch_utils.Conv1D
PyTorch 헬퍼(helper) 함수 transformers.apply_chunking_to_forward
autodoc pytorch_utils.apply_chunking_to_forward
autodoc pytorch_utils.find_pruneable_heads_and_indices
autodoc pytorch_utils.prune_layer
autodoc pytorch_utils.prune_conv1d_layer
autodoc pytorch_utils.prune_linear_layer