Store model after training

[KehanLong]

Hi,

I am trying to define new examples as you suggested by writing new control-affine systems and training. However, I did not find a way of storing the model parameters after training. (I tried to check the train_inverted_pendulum.py script but it seems that it does not store the model parameters).

Any help would be appreciated!