개발챙 AI study
/
Pytorch
/
nn.Module.register_buffer(name, tensor, persistent=True)
Search
Share
nn.Module.register_buffer(name, tensor, persistent=True)
self.register_buffer("relative_position_index", relative_position_index)
nn.Module 안에서 위와 같이 사용한다.
register_parameter와 달리 단순 buffer의 역할로,
•
optimizer로 (backpropagation으로) update되지 않는다.
•
state_dict에 저장된다 ( 다시 사용 가능 )
•
GPU에서 작동한다.