RLProject

Training models

The main script is train-model.py which allows a user to train an agent using either supervised learning or REINFORCE.

Finetuning models

To finetune a pretrained model use the script finetune.py