Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fine tuning Virtex for image captioning #19

Open
freeIsa opened this issue Dec 16, 2020 · 1 comment
Open

Fine tuning Virtex for image captioning #19

freeIsa opened this issue Dec 16, 2020 · 1 comment

Comments

@freeIsa
Copy link

freeIsa commented Dec 16, 2020

Hi there,
I am aware that Virtex used image captioning as a pretraining task and not as the "final goal", but I was wondering whether one could go on fine-tuning the pretrained model (e.g. bicaptioning_R_50_L1_H2048) with additional COCOcaptions-like data in order to get an improved captioning model.
Has anyone tried that or does anyone have any suggestion how to do it? Can any of the scripts in the repository be used/adapted for fine-tuning existing models?
Thanks a lot! :)

@bermeitinger-b
Copy link

You could use the pretrained ResNet-50 and then, similar to the pretraining layout, add the Transformers back in. But now, you freeze ResNet-50 and use more decoder layers instead of one.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants