Model Training

[JuliGH6]

Hi, I was wondering if I could train the model further with another dataset. UniGradIcon does not work too well with my data so I would like to try and train it. Who should I contact?

[lintian-a]

Hi @JuliGH6 ,

Thanks for your interest in our work!

We are preparing the training and evaluation code for uniGradICON. It will be available after we finish cleaning the code.

In the meantime, you can use the training code of GradICON. It is similar to the training code for uniGradICON except for the dataset and dataloader. Also, you can find more information in #14 regarding how to finetune uniGradICON with the above training code.

Please don't hesitate to let me know if you encounter any further issues while using GradICON's training code.

[Tara-Liu]

Hello, there is still no training code so far

[lintian-a]

Hi @Tara-Liu ,

You can find the training code at this branch