ボトムブッブとトップブロブを同じに設定することで、Caffeにメモリ消費を節約するための「インプレース」計算を行うように指示できます。Caffeでインプレースレイヤーを使用するタイミングは?
現在のところ、私が安全に使用できるのは、"BatchNorm"
、"Scale"
、"ReLU"
のレイヤーです(私が間違っている場合は教えてください)。それは他の層のためにいくつかの問題を抱えているようですが(this issue)
Caffeでインプレースレイヤーを使用するタイミングは?
バックプロパゲーションではどのように機能しますか?
奨励賞をいただきありがとうございます。それがあなたのために物事をクリアすることを願っています。 – Shai
@Shaiは答えに感謝します、それは確かに役立ちます! – dontloo