TransMorph_Transformer_for_Medical_Image_Registration icon indicating copy to clipboard operation
TransMorph_Transformer_for_Medical_Image_Registration copied to clipboard

Published 20 hours ago • verified publisher icon junyuchen245

Training problem of VoxelMorph on IXI

Open tzayuan opened this issue 2 years ago • 2 comments

Hi, @junyuchen245

I'm sorry to bother you again, but I have some questions which I can't sove it on myself.

I tried to retrain voxelmorph on IXI dataset, but I found that the dsc can't be improved with the increase of epoch.The running code for training is download in your git without any modification(except data path).

In addition, I downloaded your pretrained model and run infer.py, and the result is correct.

Here's my training log of 3 epoch:logoffice .log

Code for training:VM.zip

Thanks a lot!!

Best regards, Ztayuan

tzayuan avatar Apr 19 '22 05:04 tzayuan

Hi @tzayuan ,

It is found that the VoxelMorph script currently only compatible with PyTorch 1.7 see here. While I continue to investigate, you can test it on PyTorch 1.7 to see whether it works.

Junyu

junyuchen245 avatar Apr 19 '22 05:04 junyuchen245

Thanks for your suggestion, I will test it on PyTorch 1.7.

Hmmmm, that's sound interesting. Some low-level details may be involved, I've also very curious!!

tzayuan avatar Apr 19 '22 06:04 tzayuan