How to use pretrained weigths

[Michelotti]

Hello everyone,

I tried to look at the dedicated section to load the pretrained weigths and train them with another dataset (Pretraining with nnUNet), but I am having trouble to using -pretrained_weigths.

nnUNetv2_train TARGET_DATASET CONFIG FOLD -pretrained_weights PATH_TO_CHECKPOINT

By following the instruction above, I should insert the correct path, so for example I tried to use as argument for PATH_TO_CHECKPOINT either

/home/nnUNetFrame/dataset/Dataset114/nnUNet_results/Dataset114/nnUNetTrainer__nnUNetPlans_2d/fold_0/checkpoint_final.pth

or

/home/nnUNetFrame/dataset/Dataset114/nnUNet_results/Dataset114/nnUNetTrainer__nnUNetPlans_2d/fold_0/checkpoint_final/data.pkl

Do I make any obvious mistake to load the weigths? Thanks in advance!

Best,
Filippo

[mariakawula]

Hi, I might have run into the same problem. I get
AttributeError: 'NoneType' object has no attribute 'state_dict'
Did you manage to solve the problem and perform successful fine-tuning of a pre-trained model?

[tongchen-z]

Hi, I met with this problem too. Do you know how to fix it now? Thanks!