How to use the pretrained model?

[Dududu233]

I'm glad to find that you have open-sourced your pretrained model, I got it from google drive, but how can we just apply it to infer.py?
What do the files in the dicterory mean in your pretrained model?
Thanks for your time and reply.

[junyuchen245]

Hi @Dududu233 ,

Sorry for the late reply. You can use torch.load to load the pretrained weights (i.e., the entire .tar file) into the model. We additionally provided pre-trained weights of ViT-V-Net for IXI dataset here.

Junyu