- Python 3.6 or higher
- PyTorch version 0.1.12 or later.
Training is done via src/main.py
script. Run python -m src.main -h
from the project root directory to see its usage and all the available options.
To generate samples while not in training, use src/generation.py
script. Run python -m src.generation -h
from the project root directory to see its usage.