I want to compress my own model such as Gan's or mtcnn ,how can I change the code?can you supply a demo or document about how can we change the file to work on my own model?

[sunzhe09]

I'm trying to compress a unet-like model ,but i don't know how to edit it

[jiaxiang-wu]

If your model can be trained in an end-to-end manner, you can try to implement a new ModelHelper class, defining the data input pipepline, forward computation, and loss function, similar to those defined in the "./nets" directory. We are adding a documentation page to demonstrate how to incorporate customized models and data sets into PocketFlow.

[jiaxiang-wu]

Doc required: demonstrate how to incorporate customized models and data sets into PocketFlow.

[ginger0106]

hi, when the documents will be released?

[ginger0106]

to compress my own model, we need to implement a new ModelHelper and setup_lrn_rate, does it right? Anything else needed? Thanks

[jiaxiang-wu]

@ginger0106 @sunzhe09
We have added documentation to demonstrate how to incorporate self-defined data sets and models into PocketFlow. See here for details.