pytorchにおけるregister_bufferの役割

 

register_buffer("supports", torch.arange(Vmin, Vmax+DELTA_Z, DELTA_Z))

モジュールに永続バッファーを追加します。
これは通常、モデルパラメータと見なされるべきではないバッファを登録するために使用されます。
たとえば、BatchNormのrunning_meanはパラメーターではなく、永続的な状態の一部です。
バッファには、指定された名前を属性として使用してアクセスできます。

注:
メモリに定数を定義する必要があります。同時に、保存およびロード時にモデルを読み書きできます。

943の元の記事を公開 136のような 330,000以上を訪問

おすすめ

転載: blog.csdn.net/weixin_36670529/article/details/105188202