MASILab / 3DUX-Net

237 stars 33 forks source link

Finetuning on heart dataset with 8 classes. #57

Open gautamb2222 opened 8 months ago

gautamb2222 commented 8 months ago

I am working on a project to segment heart parts, as the data I have is limited, I finalised to finetune model on weights I got by training on FLare datset. While doing that I am having some errors attached below. Just to check if fine-tuning works fine for amos as mentioned by you, I tried finetuning with amos dataset as well, for which again I am gettting error after few iterations:

ERROR ON AMOS AFTER 131 iterations:

Screenshot 2023-11-06 at 5 59 16 PM

ERROR ON HEART DATASET, for which finetuning didn't even start:

image

I am using --dataset as feta here because feta does have the same number of classes as heart dataset(in my case) and I am expecting same transforms as for feta to work for my case as well. I tried changing transforms but can't get this thing right and leading to more new errors changing anything.

Please suggest me something which might be helpful for me, I am just trying to get a baseline from your implementation.