Solved by Abhilash_Srivastava in post #4 
              
              
                I think this can be the problem: 
        self.gru_rating = tnn.GRU(input_size = 50, hidden_size = 100, num_layers = 5, bidirectional = True)
        self.gru_buscat = tnn.GRU(input_size = 50, hidden_size = 100, num_layers = 5, bidirectional = True)
        self.fc_rating = tnn.Linear(100, 1)
      …