search

alasdairtran / transform-and-tell

[CVPR 2020] Transform and Tell: Entity-Aware News Image Captioning
https://transform-and-tell.ml/
89 stars 14 forks source link

other news domains #13

Open monajalal opened 3 years ago

monajalal commented 3 years ago

Great job. Is there a way to reproduce the results for other news outlets besides new york times without training from scratch? How can we fine-tune your trained model on our own news dataset?

Basically, I want to run your model to make predictions on my news dataset.

https://transform-and-tell.ml/

alasdairtran commented 3 years ago

Send me an email to request the pretrained model. You would then need to write a custom DatasetReader that takes your news data (e.g. this is my nytimes reader), along with a custom config file (example). You probably need to familiarize yourself with the AllenNLP framework before you can make changes to the code, as most of the training and evaluation code is built on top of it.