I changed that line to:

x_out = self.out(x_up)

and now for 1 epoch it’s predicting, but for the next epochs, it’s not. and the thing is I’m printing the loss, and it’s showing that the loss is getting lower and lower.