Usage¶

Training¶

Set a training configuration in the config.py script. Each parameter is commented. See the documentation file for further info about each specific hyperparameter. You can also specify the parameters when calling the main.py script following the syntax Key=Value
Train!:
```
python main.py
```

Decoding¶

Once we have our model trained, we can translate new text using the caption.py script. If we want to use the models from the first three epochs to translate the test split from our dataset, just run:

python caption.py --models trained_models/tutorial_model/epoch_1 \
                                   trained_models/tutorial_model/epoch_2 \
                          --dataset datasets/Dataset_tutorial_dataset.pkl \
                          --split test