torch.nn.Module.register_buffer の挙動確認メモ - Qiita
PyTorch で torch.nn.Module を継承したモデルに torch.nn.Module.register_buffer によって登録したテンソルは、勾配減少方向への更新は受けないが、モデルの状態としてセーブ&ロードされます。以下の手順でそれを確認できます。 ...