Can you please release a pretrained model?

[xiaomeixivvr]

This is a good work!
I like it.
If you can release a pretrained model, thanks.

[maria-larchenko]

hi! please check the hugginface link
https://huggingface.co/MariaLarchenko/modflows_color_encoder/tree/main

There are three checkpoints, B6 (EfficientNet B6) is the main model and B0 (EfficientNet B0) was used for ablation study.
To use these models, refer to the demo notebook
https://github.com/maria-larchenko/modflows/blob/main/ModFlows_demo.ipynb