- pytorchのadd_moduleを使用し、任意の場所に層を追加しいたいです。

実現したいこと

  • pytorchのadd_moduleを使用し、画像の①の場所に層を追加しいたいです。

発生している問題・エラーメッセージ

add_moduleを使用すると②の場所に追加されてしまいます。ネットで調べたのですが、モデルの最初に追加しているものなどはなく、実装できずに困っています。どのように行えばよいでしょうか。

イメージ説明

コメントを投稿

0 コメント