learned_primal_dual icon indicating copy to clipboard operation
learned_primal_dual copied to clipboard

Published 20 hours ago • verified publisher icon adler-j

NAN loss when run 'mayo_learned_primal_dual.py'

Open wangwei-cmd opened this issue 3 years ago • 1 comments

Hi, I can run 'mayo_learned_primal_dual.py', but the loss is always NAN during the training process. I also run 'learned_primal_dual.py' and the loss seems to be right. Is there a probelm with my data or other reasons? Thank you very much.

wangwei-cmd avatar Jun 10 '21 04:06 wangwei-cmd

hi. i had same problem. you should be created your own database. and then changed ndata to example 1000(all of your .npy array number) to solve this problem

hjahan58 avatar Jun 10 '21 12:06 hjahan58