about the model save and test code

[Danee-wawawa]

Hi, thank you for your work. How to save the generated model? Can you provide the test code?
Looking forward to your reply.
@tnarek @shaibagon