Features become all Nan after self.rnn layer, and loss is nan. 