nn.ReLUにおけるinplaceの役割(inplace = True)

ドキュメントの説明は次のとおりです。

  • パラメーター:上書き操作を実行するかどうかのインプレース選択

これは、取得した値によって計算された以前の値を上書きするかどうかを意味します。次に例を示します。

x = x +1

つまり、元の値を操作してから、取得した値を値に直接コピーします

上書き操作の代わりの例は次のとおりです。

y = x + 1

x = y

このように、もう1つの変数yを格納するためにメモリを使用する必要があるため、

nn.Conv2d(3, 64, kernel_size=11, stride=4, padding=2),
nn.ReLU(inplace=True)

その意味は、上位ネットワークConv2dから渡されたテンソルを直接変更することです。これにより、コンピューティングメモリを節約でき、他の変数を保存する必要がなくなります。

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

おすすめ

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