https://github.com/tchaton/lightning-hydra-seed
Pytorch Lightning seed project with hydra
https://github.com/tchaton/lightning-hydra-seed
Last synced: 9 months ago
JSON representation
Pytorch Lightning seed project with hydra
- Host: GitHub
- URL: https://github.com/tchaton/lightning-hydra-seed
- Owner: tchaton
- License: mit
- Created: 2020-10-07T11:21:14.000Z (over 5 years ago)
- Default Branch: main
- Last Pushed: 2020-10-08T17:38:20.000Z (over 5 years ago)
- Last Synced: 2024-12-30T07:51:23.271Z (over 1 year ago)
- Language: Python
- Size: 1.57 MB
- Stars: 18
- Watchers: 2
- Forks: 2
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Pytorch Lightning seed project with hydra
[](https://codecov.io/gh/tchaton/lightning-hydra-seed) [](https://github.com/tchaton/lightning-hydra-seed/actions)
### Setup
```
pyenv install 3.7.8
pyenv local 3.7.8
python -m venv .venv
source .venv/bin/activate
python -m pip install -U pip
python -m pip install poetry
python -m poetry install
```
### PRINCIPAL CMD
```python
python train.py model={{MODEL}} dataset={{DATASET}} loggers={{LOGGERS}} log={{LOG}} notes={{NOTES}} name={{NAME}} jit={{JIT}}
```
- `LOGGERS` str: Configuration file to log to Wandb, currently using mine as `thomas-chaton`
- `LOG` bool: Wheter to log training to wandb
- `NOTES` str: A note associated to the training
- `NAME` str: Training name appearing on Wandb.
- `LOG` bool: Wheter to make model jittable.
### Current Demo Command
```python
python train.py task=categorical_classification model=simple_mlp dataset=mnist loggers=thomas-chaton log=False
```
```python
python train.py task=categorical_classification model=simple_mlp dataset=mnist loggers=thomas-chaton log=False notes="My Simple MLP with Adam" notes="My Simple MLP with Adam" name="My Simple MLP with Adam" jit=False
```
### Register your project on [codecov](https://codecov.io)
Set your secret key in settings with `CODECOV_TOKEN`