## Requirements

- Python 3
- PyTorch

## Install

Install the required packages in the `requirements.txt` by: `pip install requriements.txt`

## Training

To train the model, run:
```
bash train.sh model_name
```

Model checkpoints and logs will be saved to `./saved_models/model_name`.

For the complete list of parameters see `train.py`.

## Evaluation

To run evaluation on the test set, run:
```
python eval.py saved_models/model_name --dataset test
```

This will use the `best_model.pt` file by default. Use `--model checkpoint_epoch_10.pt` to specify a model checkpoint file.
