Replace pre-trained weights for the Primal-Dual example (ckp_PDNet.pth)
#6
by PAUL-BERNARD - opened
For some reason, current pre-trained weights give very bad reconstructions on Learned Primal-Dual algorithm for CT scan (https://deepinv.org/auto_examples/unfolded/demo_learned_primal_dual.html#sphx-glr-auto-examples-unfolded-demo-learned-primal-dual-py).
The weights I propose are trained on the Primal-Dual example notebook, with 9 epochs.