層(Layer)の名前で層の出力変数を参照したいのですが

下記のようにNNを構造する時に名前を付けていますが、
NNのinstance(object)を生成した後、これらの名前でその層の出力を参照したいですけれども、方法を教えてください。

python

self.layers.add_module("Conv2d", self.conv )self.layers.add_module("BatchNorm", nn.BatchNorm2d(outPlanes))

どうぞよろしくお願いします。

コメントを投稿

0 コメント