pytorch_template

Requirements

• Python 3.10
• Poetry

Usage

Prepare

sed -i 's/pytorch_template/<your-project-name>/g' pyproject.toml
mv -rf pytorch_template <your-project-name>

poetry install

And then, edit config files in conf/ directory.

Train

poetry run python train.py

Test

poetry run python test.py --model-path <path-to-checkpoint>