Open TousenKaname opened 8 months ago
And I found that 'args.use_pseudo_label' only affect the loss calculate, it means we also have to dounsupervised_forward
method, consuming computing waste...
I'm sorry for that. The args.use_pseudo_label is still unfinished. The original released code only supports training with pseudo labels.
But you can disable the pseudo-label training by discarding all the code related to the pseudo-label in the training process. SegVol will perform better training with only ground truth labels in specific categories.
Thanks for your excellent work! But I encounter a bug when I set the 'use_pseudo_label=False'.
The loss is not defined...