search

XiaoMi / mace-models

Mobile AI Compute Engine Model Zoo
Apache License 2.0
368 stars 72 forks source link

Please provide training scripts (or at least more details) for realtime-style-transfer model. #22

Closed ljoukov closed 4 years ago

ljoukov commented 4 years ago

README for https://github.com/XiaoMi/mace-models/tree/master/realtime-style-transfer says that: "This realtime style transfer model was compressed and optimized from here (fast style transfer project)."

Could you please provide more details what exactly was done and share training script source code? Thanks!

TonyMou commented 4 years ago

We used a step-traning method to train this model, and the step-training codes are here(https://github.com/TonyMou/fast-style-transfer/tree/support-step-training).