Fixed a crash in load_best_model_for_inference

Amshaker / unetr_plus_plus

[IEEE TMI-2024] UNETR++: Delving into Efficient and Accurate 3D Medical Image Segmentation

Apache License 2.0

360 stars 34 forks source link

Fixed a crash in load_best_model_for_inference #51

Open korenminchev opened 1 year ago

korenminchev commented 1 year ago

Fixed a crash where the checkpoint folder wasn't being passed on in load_model_for_inference. This caused a crash in restore_model because folder was None.