We provide the pretraining script as well as the pretrained model to support the finetuning on downstream tasks. More models will be updated continuously.
The following script run the pretraining on the MSCOCO, VG, SBU, Conceptual datasets.
$ bash scripts/train-pt.sh
As the pretraining stage complete, you may run the finetuning scripts on downstream tasks.
name | model size | download |
---|---|---|
ROSITA-base | 116M | model |