Training
PyTorch Lightening
https://pytorch-lightning.readthedocs.io/en/latest/
Hyperparameter Optimization
Optuna
Optuna - A hyperparameter optimization framework
Argument Parsing
Hydra
- configurations (replaces
argparse) - https://hydra.cc/
- youtube: # Configuration Management For Data Science Made Easy With Hydra
Argparse
https://docs.python.org/3/library/argparse.html
Logging
- TensorBoard
- Wandb
Human Learn
https://koaning.github.io/human-learn/index.html
Nvidia-htop
https://github.com/peci1/nvidia-htop
Nvitop
https://rasbt.github.io/mlxtend/
pipreqs
this package extracts requirements.txt file.
Python libraries to use
Pathlib
to read a write to a file
Pillow (a PIL fork)
to work with images
Use this one instead of opencv
Convert from different dataset formats
https://github.com/pylabel-project/pylabel
Albumentations
Work with images and bounding boxes and augment them. I used it for resize the images.
GPU stuck problem
Killall -u pourmand