Training code
EricYu97 opened this issue · comments
Hello, I'm very impressed to see the results of your maxim model and want to train with my dataset. But I do not see any training code in this repository. Would you like to commit the training code? Thanks!
Hi we don't have any plans to release the training code on Jax, unfortunately.
But we plan to reimplement it in PyTorch and provide training code. Please keep an eye on maxim-pytorch