Train my own model using maxim.py

google-research / maxim

[CVPR 2022 Oral] Official repository for "MAXIM: Multi-Axis MLP for Image Processing". SOTA for denoising, deblurring, deraining, dehazing, and enhancement.

https://arxiv.org/abs/2201.02973

Apache License 2.0

977 stars 105 forks source link

Train my own model using maxim.py #17

Closed zarmondo11 closed 2 years ago

zarmondo11 commented 2 years ago

Hello and thank you for your insane paper... I have a question... How can I train my own model instead of using checkpoints?

vztu commented 2 years ago

We don't have any plans to release the Jax training pipeline, unfortunately. However, we plan to reproduce the model in PyTorch and release the training/validation code here. The repo is still under construction, but I am busy with graduation stuff recently, so please stay tuned...

zarmondo11 commented 2 years ago

Thank you but I will ask another question in pytorch rep.