I have been using more complex models ( rnns, deconv layers, cnn,etc) in various configurations.  I’ve tried to reduce the model to the simplest complexity as possible. In this current repo, the final layer I use  is a linear layer that  takes an embedding layer and outputs 40x40pixels x 3 (rgb channels). I’m not sure what I am doing that is wrong.