Bumped version for release

Author: gperdrizet

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "image-classification-tools"
-version = "0.5.7"
+version = "0.5.8"
 description = "A lightweight PyTorch toolkit for building and training image classification models"
 authors = ["gperdrizet <[email protected]>"]
 readme = "src/image_classification_tools/README.md"