Actually, the model was saved  on nn.Sequential form, so I can load them when I defined the model as nn.Sequential.
Is there any simple way to save the parameters so that model defined above can load them?
Thanks.