scil-vital / dwi_ml

Code for people working on machine learning for diffusion MRI
MIT License
8 stars 7 forks source link

Improve visu of loss #225

Closed EmmaRenauld closed 9 months ago

EmmaRenauld commented 9 months ago

Visu of loss: print range, show histogram.

Add tqdm for nicer loop.

Add option to use last epoch rather than best model.