下記のようにNNを構造する時に名前を付けていますが、NNのinstance(object)を生成した後、これらの名前でその層の出力を参照したいですけれども、方法を教えてください。
python
self.layers.add_module("Conv2d", self.conv )self.layers.add_module("BatchNorm", nn.BatchNorm2d(outPlanes))
どうぞよろしくお願いします。
0 コメント