Skip to content

Latest commit

 

History

History
44 lines (31 loc) · 631 Bytes

README.md

File metadata and controls

44 lines (31 loc) · 631 Bytes

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/