airlab-unibas / airlab

Image registration laboratory for 2D and 3D image data
Apache License 2.0
408 stars 92 forks source link

Could you add a .loss() function to the registration classes? #3

Closed bsugerman closed 6 years ago

bsugerman commented 6 years ago

While it is helpful to have the torch optimizer return the loss as it iterates, I think it would be VERY helpful to provide a .loss() function to the registration class, so that the user can print out the current loss value once the registration is finished. Seems like it could be an easy call to the self._optimizer?

bsugerman commented 6 years ago

Never mind. I did it in my fork and will push that soon.