- Inspiration: Kaggle Digit Recognizer
From the source directory run the following commands
python3 -m venv venv
for initialising the virtual environmentsource venv/bin/activate
for activating the virtual environment
The following commands shall be ran after activating the virtual environment.
pip install --upgrade pip
for upgrading the pippip install -r requirements.txt
for the functional dependenciespip install -r requirements-dev.txt
for the development dependencies. (should includepre-commit
module)pre-commit install
for installing the precommit hook
For the extra modules, which are not a standard pip modules (either from your own src
or from any github repo)
pip install -e .
for the files/modules insrc
to be accessed as a package. This is accompanied withsetup.py
andsetup.cfg
files-e
means installing a project in editable mode, thus any local modifications made to the code will take effect without reinstallation.
Model | On the Validation set | On Kaggle set |
---|---|---|
Basic Conv | 100% | 99% |
Mobilenet | xx.xx% | xx.xx% |
VGG16 | xx.xx% | xx.xx% |
Resnet50 | 99.00% | 98.85% |
Resnet18 | 99.19% | 99.17% |
WideResnet28-10 | xx.xx% | xx.xx% |
- To activate Tensorboard:
tensorboard --logdir=model_chkpt/lightning_logs/
- To train a model:
python ./src/train.py -m <model_name> -e <epochs> -bs <batch_size> -lr <learning_rate> -v <version>
- Unweighted average
- Majority voting
- Super Learner
pytest --durations=0 #Show all times for tests and setup and teardown
pytest --durations=1 #Just show me the slowest