我想把一个vgg网络的参数传递给一个风格迁移用的vgg网络,但是两个网络的结构不一致(原网络有一些ReLU和Pooling层)。我想要仅仅转移所有卷积层的参数,但是vgg.children()的方法会返回网络中所有的层,有没有方法可以让我只提取其中的卷积层?或者有没有什么更加便利的写法让我加载参数?
相似问题