Informal implementation from scratch for educational purpose.
- LeNet
- AlexNet
- MNIST
- CalTech101
- CIFAR 10
- PyTorch
- PyTorch Lightning
- Use
pip-compile
to update Python requirements files. Can be installed with:python3 -m pip install pip-tools
. - Use
pre-commit
hooks to automate static checks:
pip install pre-commit
pre-commit install
pre-commit install --install-hooks
Important formula to compute the size of the output tensor: $$ \frac{W - K + 2P}{S} + 1 $$ With
-
$W$ the input volume -
$K$ the filter size -
$P$ the padding size -
$S$ the stride size