PyTorch: Adding hooks to model for saving the intermediate layers output returns the features two times

后端 未结 0 1791
星月不相逢
星月不相逢 2021-01-23 19:37

I want to implement the Style Transfer paper using Pytorch and the VGG19 network. For this, I need the intermediate output features for some layers. I named the convolutional mo

相关标签:
回答
  • 消灭零回复
提交回复
热议问题