Suggestion for a Complete Training Framework for Style Transfer Models

[cmgzn]

Hello,

I recently started looking the library, and I had the impression that it provides a comprehensive framework for developers to train their own style transfer models with a certain number of articles. However, I found that the library currently only includes four pre-trained style transfer models created by the author, along with some example datasets.

I believe that implementing a training framework would be much more meaningful and beneficial for users who want to customize their models. This would allow developers to easily train their own models based on their specific data, enhancing the library's usability and functionality.

Have you considered creating this project according to this idea? If you are thinking about it, I would be more than happy to offer my assistance in my spare time.