# Train captioning models

This directory contains scripts for training captioning model

## Train state based captioning model
First fill in the directory for where the data are in the script, then run:
```
python train_captioning_model.py
```

## Train image based captioning model
First fill in the directory for where the data are in the script, then run:
```
python train_captioning_model_image.py
```
