Solved by ptrblck in post #6 
              
              
                Thanks for the data sample. 
It looks like you are dealing with some kind of indices, is that correct? 
If that’s the case, I would recommend to use an nn.Embedding layer to map these indices to some dense floating point representation. 
Here is a small code modification to your model: 

class revie…