Is "post pre-training (Imagenet weight init) on custom dataset" possible with this git code?

google-research / big_transfer

Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.

https://arxiv.org/abs/1912.11370

Apache License 2.0

1.5k stars 175 forks source link

Is "post pre-training (Imagenet weight init) on custom dataset" possible with this git code? #83

Open flipflop98 opened 1 year ago

flipflop98 commented 1 year ago

To whom it may relate:

As title shows, I hope to pretrain on my dataset with this big_transfer code. But I'm not sure whether it is possible becuase the git seems to support fine-tune only. Any comments welcome.

Thank you in advance.

Changhyun Kim