Currently when I use
torch.save(lstm_net.state_dict())

it does not save custom parameters I have added to the module. It only saves the weights and biases of the lstm and dense layers.
How do I fix it so it does?