Open yy042 opened 1 month ago
Hi @yy042 , can you please share all the commands you used when following the pretraining steps?
One thing you could try is setting your checkpoint path in quotation marks like this:
nnUNetv2_train TARGET_DATASET CONFIG FOLD -pretrained_weights PATH_TO_CHECKPOINT, as I run nnUNetv2_train 20 3d_fullres 0 -pretrained_weights "D:\nnUNet\nnUNetFrame\nnUNet_results\Dataset040_KiTS19\nnUNetTrainer__custom_plan1__3d_fullres\fold_all\checkpoint_best.pth"
Best, Sebastian
Thank you for your reply! I've tried setting checkpoint path in quotation marks, but the same error occurred again. Here are the commands I used when following the pretraining tutorial:
source dataset
-Dataset040_KiTS19
target dataset
-Dataset020_KidneyTumor
nnUNetv2_plan_and_preprocess -d 20
nnUNetv2_extract_fingerprint -d 40
nnUNetv2_move_plans_between_datasets -s 20 -t 40 -sp nnUNetPlans -tp custom_plan1
nnUNetv2_preprocess -d 40 -plans_name custom_plan1
nnUNetv2_train 40 3d_fullres all -p custom_plan1
nnUNetv2_train 20 3d_fullres 0 -pretrained_weights "D:\nnUNet\nnUNetFrame\nnUNet_results\Dataset040_KiTS19\nnUNetTrainer__custom_plan1__3d_fullres\fold_all\checkpoint_best.pth"
-AssertionError: The requested nnunet trainer class must inherit from nnUNetTrainerHi @yy042 ,
can you try running the train command when adding -tr nnUNetTrainer
?
Hello. Thank you for your amazing work. I got a problem when I try to fine tune a pretrained model on my personal dataset(Dataset020). Following the steps in documentation/pretraining_and_finetuning.md, I have finished pretraining on Dataset040_KiTS19 successfully. But when I try to run the last order
nnUNetv2_train TARGET_DATASET CONFIG FOLD -pretrained_weights PATH_TO_CHECKPOINT
, as I runnnUNetv2_train 20 3d_fullres 0 -pretrained_weights D:\nnUNet\nnUNetFrame\nnUNet_results\Dataset040_KiTS19\nnUNetTrainer__custom_plan1__3d_fullres\fold_all\checkpoint_best.pth
, an error occured:It's odd because I've been strictly follow the instructions in documentation/pretraining_and_finetuning.md, and I have never changed the structure of my nnU-Net. I would appreciate it if you could point out what I have done is wrong. Thank you for your support!