2023年3月12日
取出预训练模型中间层的输出(pytorch)
摘要： 1 遍历子模块直接提取 对于简单的模型，可以采用直接遍历子模块的方法，取出相应name模块的输出，不对模型做任何改动。该方法的缺点在于，只能得到其子模块的输出，而对于使用nn.Sequensial()中包含很多层的模型，无法获得其指定层的输出。 示例 resnet18取出layer1的输出 from 阅读全文
posted @ 2023-03-12 09:48 湾仔码农 阅读(2632) 评论(0) 推荐(0) 编辑