I am trying to train a PyTorch LSTM model which is defined as:
class RecurrentNet(torch.nn.Module): def __init__(self, d_in, d_hidden, sequence_length):