Add ability to set multiple testing / validation loss functions, so don't have to call solver.test twice

CambridgeCIA / LION

Learned Iterative Optimization Networks

GNU General Public License v3.0

19 stars 14 forks source link

Open cshoebridge opened 3 months ago

cshoebridge commented 3 months ago

Just seems like a nice QOL feature that won't be too hard / offensive to implement

hyt35 commented 3 months ago

see my pull request, reconstructor class allows you to pass metrics, current default is SSIM and PSNR