Closed Haalum closed 10 months ago
As FSC-147 is a quite small dataset, it will be hard to train a good feature extractor without any kind of pre-training.
But if you want to try it, you should also remove this line which freezes the image encoder, besides what you have done.
For the epoches, maybe the larger the better? But I am not quite sure whether it will work.
I understand. I'm just curious. Thank you.
FSC_finetune_cross.py
.misc.load_model_FSC(args=args, model_without_ddp=model_without_ddp)
Am I missing anything?
Thanks!