Lightning Hydra Template

UV

UV_EXTRA_INDEX_URL=https://download.pytorch.org/whl/cpu uv sync

Train

Dev Run

python src/train.py experiment=catdog_ex +trainer.fast_dev_run=True

python src/train.py experiment=catdog_ex +trainer.log_every_n_steps=5

Docker

docker build -t lightning-hydra .

docker run -it --rm \
	-v `pwd`/data:/app/data \
	-v `pwd`/logs:/app/logs \
	lightning-hydra \
	python src/train.py experiment=catdog_ex

Coverage

coverage run -m pytest
coverage report -m

Github Actions

Test locally with https://nektosact.com/