pytorch提取某一层的特征图

直接在forward函数里，return output_con2.data

如：

def forward():

...

 base_out = self.base_model(input_var)
 output = self.new_fc(base_out)
 return output
想要输出base_out

只需要加入 out_512=base_out.data

如：

base_out = self.base_model(input_var)

out_512=base_out.data
output = self.new_fc(base_out)

return output

如果out_512=base_out.data不加data，out_512的requires_grad=True,

在训练或测试的过程中，GPU得不到释放，出现out memory问题