aCASTLE icon indicating copy to clipboard operation
aCASTLE copied to clipboard

Training without pretrained weights

Open chinmayembedded opened this issue 4 years ago • 1 comments

Hi,

I trained training aCASTLE by 1 shot 5 way strategy without using pretrained weights. I trained model for 80 epochs. Performance numbers observed were drastically low compared to training with pretrained weights.

If I want to train a model without pretrained weights, is it possible? or Can you please share methodology to create pretrained weights files.

Thank you!

chinmayembedded avatar May 06 '21 16:05 chinmayembedded

Hi, we pre-train the model with the same strategy as in the FEAT repo. So you may try the pre-trained weights provided by that repo.

Based on our observation, we find the model with pre-trained weights gets much higher GFSL performance than the one without pre-trained weights, but they have similar FSL performance.

BTW, if training from scratch, it may require much longer epochs.

Han-Jia avatar May 10 '21 08:05 Han-Jia