Hello,
Thank you for the great work and sharing the code!
I am sorry to ask where is the training code, if it is shared. By the way, I find there is a optimizer.step() for the validation set in convex_adam_MIND.py, I am not sure if it is the training code.
Thankyou
Best wishes