Train script
Hi Eduardo,
First of all, thank you for publishing all of these models trained on smaller datasets, this really helps setting up experiments quickly !
I am trying to train from scratch some of you models, only changing the seed.
Is there a simple way to achieve this by using the hyperparameters you provide in hyperparameters.json (using the Trainer class maybe)? Or did you use a custom training script ?
Cheers,
Augustin
Hi Augustin,
I’m glad you find them useful. The custom training script with the dependencies can be found here:
https://github.com/edadaltocg/detectors/blob/master/scripts/train.py
Otherwise, you should be able to reproduce it with the HF’s Trainer class by adapting a little bit the code.
Best,
Eduardo
Thanks a lot!
Hi Eduardo,
I have an other quick question. In the training script, there is a --pretrained flag.
Was the --pretrained set to train the vit_base_patch16_224_in21k_ft_cifar10 model?
Since there are only 10 epochs in the config I guess it is the case, but I just want to be sure.
Best,
Augustin