NAN loss when run 'mayo_learned_primal_dual.py'

[wangwei-cmd]

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.

[hjahan58]

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