take the output from a specific layer in pytorch

后端 未结 0 958
日久生厌
日久生厌 2021-02-04 12:01

I have implemented an autoencoder in Pytorch and wish to extract the representations (output) from a specified encoding layer. This setup is similar to making predictions using

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