# Usage
This is a simple implementation. You only need modify the `trainer.py` file.
All hyper-parameters and paths are in the `__init__` function of the **Trainer**.

## Train
- Assign the training dataset path to the `self.train_dataset_dir`
variable in the `__init__` function.

- Uncomment training code in the main function.

## Test
- Assign the benchmark dataset path to the `self.benchmark_dir`
variable in the `__init__` function.

- Uncomment testing code in the main function.

## Inference 
- Uncomment inference code in the main function.
