search

DeepRegNet / DeepReg

Medical image registration using deep learning
Apache License 2.0
564 stars 76 forks source link

new naming scheme for the saved trained models: prediction script needs to be modified accordingly #641

Closed ciphercharly closed 3 years ago

ciphercharly commented 3 years ago

the prediction scripts that were expecting something like (this from the paired brain demo)

ckpt_path = f"{log_root}/dataset/pretrained/learn2reg_t1_paired_train_logs/save/weights-epoch800.ckpt"

need to be updated now to the new naming scheme

the saved checkpoints are now like

save/ckpt-****.data-00000-of-00001
save/ckpt-****.index

so /save/ckpt-***** would be the new name in the prediction script

ciphercharly commented 3 years ago

(lol, sorry for this useless issue, but I was getting an error in prediction and I figured out it was the new naming scheme when I had already opened the issue)

mathpluscode commented 3 years ago

Yes, we changed the ckpt system to enable restoring from a stopped run. 😬

ciphercharly commented 3 years ago

that's great, very useful

mathpluscode commented 3 years ago

that's great, very useful

Please let us know if it works as expected 😸 it should.