OpenSeq2Seq
OpenSeq2Seq copied to clipboard
Change the shape of FC layer (vocab size)
Hi everyone! I want to change the size of fully connected layers of a pre-trained model. In other words, I want to replace fully connected layers with other fully connected layers with different shapes.
Take a look at https://nvidia.github.io/OpenSeq2Seq/html/sentiment-analysis.html example