airlab-unibas / airlab

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

How to get final loss value? #6

Closed adizhol closed 5 years ago

RobinSandkuehler commented 5 years ago

Hi, you can get the final loss value by calling

final_loss_value = registration.loss.data.item()

after the registration.