Skip to content

elbayadm/captioning

Repository files navigation

Captioning code in PyTorch

Building from Ruotian Luo's code for captioning

Data preprocessing:

> python scripts/prepare/prepro.py 

Training:

Training requires some directories for saving the model's snapshots, the tensorboard events

> mkdir -p save events

To train a model under the parameters defined in config.yaml

> python train.py -c config.yaml 

Check options/opts.py for more about the options.

To evaluate a model:

> python eval.py -c config

To submit jobs via OAR use either train.sh or select_train.sh