fix: Fix wrong automatic batch size implementation coming from lightn…

…ing (#118) * fix: Fix wrong automatic batch size implementation coming from lightning * build: Upgrade version * docs: Update changelog
orobix · May 28, 2024 · 508406d · 508406d
1 parent 112b405
commit 508406d
Show file tree

Hide file tree

Showing 4 changed files with 388 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,12 @@
 # Changelog
 All notable changes to this project will be documented in this file.
 
+### [2.1.7]
+
+#### Fixed
+
+- Fix lightning implementation of batch size finder not working properly when the initial batch size is bigger than the dataset length, now also the code checks that the last iteration works properly.
+
 ### [2.1.6]
 
 #### Updated

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "quadra"
-version = "2.1.6"
+version = "2.1.7"
 description = "Deep Learning experiment orchestration library"
 authors = [
 	"Federico Belotti <[email protected]>",

diff --git a/quadra/__init__.py b/quadra/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "2.1.6"
+__version__ = "2.1.7"
 
 
 def get_version():