quadra 2.3.0a1__tar.gz → 2.3.0a3__tar.gz

{quadra-2.3.0a1 → quadra-2.3.0a3}/PKG-INFO

@@ -1,19 +1,18 @@
 Metadata-Version: 2.1
 Name: quadra
-Version: 2.3.0a1
+Version: 2.3.0a3
 Summary: Deep Learning experiment orchestration library
 Home-page: https://orobix.github.io/quadra
 License: Apache-2.0
 Keywords: deep learning,experiment,lightning,hydra-core
 Author: Federico Belotti
 Author-email: federico.belotti@orobix.com
-Requires-Python: >=3.9,<3.11
+Requires-Python: >=3.10,<3.11
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Education
 Classifier: Intended Audience :: Science/Research
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development
@@ -21,7 +20,7 @@ Classifier: Topic :: Software Development :: Libraries
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Provides-Extra: onnx
 Requires-Dist: albumentations (>=1.3,<1.4)
-Requires-Dist: anomalib-orobix (==0.7.0.dev143)
+Requires-Dist: anomalib-orobix (==0.7.0.dev150)
 Requires-Dist: boto3 (>=1.26,<1.27)
 Requires-Dist: grad-cam-orobix (==1.5.3.dev001)
 Requires-Dist: h5py (>=3.8,<3.9)
@@ -32,6 +31,7 @@ Requires-Dist: label_studio_converter (>=0.0,<0.1)
 Requires-Dist: matplotlib (>=3.6,<3.7)
 Requires-Dist: minio (>=7.1,<7.2)
 Requires-Dist: mlflow-skinny (>=2.3.1,<3.0.0)
+Requires-Dist: numpy (<2)
 Requires-Dist: nvitop (>=0.11,<0.12)
 Requires-Dist: onnx (==1.15.0) ; extra == "onnx"
 Requires-Dist: onnxconverter-common (>=1.14.0,<2.0.0) ; extra == "onnx"
@@ -39,10 +39,10 @@ Requires-Dist: onnxruntime_gpu (==1.20.0) ; extra == "onnx"
 Requires-Dist: onnxsim (==0.4.28) ; extra == "onnx"
 Requires-Dist: opencv_python_headless (>=4.7.0,<4.8.0)
 Requires-Dist: pandas (<2.0)
-Requires-Dist: pillow (>=9.3,<9.4)
-Requires-Dist: pydantic (>=1.10.10)
+Requires-Dist: pillow (>=10,<11)
+Requires-Dist: pydantic (==1.10.10)
 Requires-Dist: python_dotenv (>=0.21,<0.22)
-Requires-Dist: pytorch_lightning (>=2.1,<2.2)
+Requires-Dist: pytorch_lightning (>=2.4,<2.5)
 Requires-Dist: rich (>=13.2,<13.3)
 Requires-Dist: scikit_learn (>=1.2,<1.3)
 Requires-Dist: scikit_multilearn (>=0.2,<0.3)
@@ -50,11 +50,11 @@ Requires-Dist: seaborn (>=0.12,<0.13)
 Requires-Dist: segmentation_models_pytorch-orobix (==0.3.3.dev1)
 Requires-Dist: tensorboard (>=2.11,<2.12)
 Requires-Dist: timm (==0.9.12)
-Requires-Dist: torch (==2.1.2)
+Requires-Dist: torch (==2.4.1)
 Requires-Dist: torchinfo (>=1.8,<1.9)
 Requires-Dist: torchmetrics (>=0.10,<0.11)
 Requires-Dist: torchsummary (>=1.5,<1.6)
-Requires-Dist: torchvision (>=0.16,<0.17)
+Requires-Dist: torchvision (>=0.19,<0.20)
 Requires-Dist: tripy (>=1.0,<1.1)
 Requires-Dist: typing_extensions (==4.11.0) ; python_version < "3.10"
 Requires-Dist: xxhash (>=3.2,<3.3)

{quadra-2.3.0a1 → quadra-2.3.0a3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "quadra"
-version = "2.3.0a1"
+version = "2.3.0a3"
 description = "Deep Learning experiment orchestration library"
 authors = [
 	"Federico Belotti <federico.belotti@orobix.com>",
@@ -38,12 +38,13 @@ build-backend = "poetry.core.masonry.api"
 quadra = "quadra.main:main"
 
 [tool.poetry.dependencies]
-python = ">=3.9,<3.11"
+python = ">=3.10,<3.11"
 
-torch = { version = "2.1.2", source = "torch_cu121" }
-torchvision = { version = "~0.16", source = "torch_cu121" }
+torch = { version = "2.4.1", source = "torch_cu121" }
+torchvision = { version = "~0.19", source = "torch_cu121" }
 
-pytorch_lightning = "~2.1"
+pytorch_lightning = "~2.4"
+numpy = "<2"
 torchsummary = "~1.5"
 torchmetrics = "~0.10"
 hydra_core = "~1.3"
@@ -53,13 +54,13 @@ mlflow-skinny = "^2.3.1"
 boto3 = "~1.26"
 minio = "~7.1"
 tensorboard = "~2.11"
-pillow = "~9.3"
+pillow = "^10"
 pandas = "<2.0"
 opencv_python_headless = "~4.7.0"
 python_dotenv = "~0.21"
 rich = "~13.2"
 scikit_learn = "~1.2"
-pydantic = ">=1.10.10"
+pydantic = "1.10.10"
 grad-cam-orobix = "1.5.3.dev001"
 matplotlib = "~3.6"
 seaborn = "~0.12"
@@ -72,7 +73,7 @@ h5py = "~3.8"
 timm = "0.9.12"
 
 segmentation_models_pytorch-orobix = "0.3.3.dev1"
-anomalib-orobix = "0.7.0.dev143"
+anomalib-orobix = "0.7.0.dev150"
 xxhash = "~3.2"
 torchinfo = "~1.8"
 typing_extensions = { version = "4.11.0", python = "<3.10" }
@@ -88,6 +89,7 @@ name = "torch_cu121"
 url = "https://download.pytorch.org/whl/cu121"
 priority = "explicit"
 
+
 [tool.poetry.group.dev]
 optional = true
 
@@ -95,12 +97,14 @@ optional = true
 hydra-plugins = { path = "quadra_hydra_plugin" }
 # Dev dependencies
 interrogate = "~1.5"
-pre_commit = "~3.0"
+pre_commit = "^3.0"
 pylint = "~2.16"
 types_pyyaml = "~6.0.12"
 mypy = "^1.9.0"
 pandas_stubs = "~1.5.3"
 twine = "~4.0"
+ipython = ">8"
+ipykernel = ">6"
 
 
 # Test dependencies

{quadra-2.3.0a1 → quadra-2.3.0a3}/quadra/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.3.0a1"
+__version__ = "2.3.0a3"
 
 
 def get_version():

{quadra-2.3.0a1 → quadra-2.3.0a3}/quadra/configs/callbacks/all.yaml

@@ -30,3 +30,16 @@ progress_bar:
 lightning_trainer_setup:
   _target_: quadra.callbacks.lightning.LightningTrainerBaseSetup
   log_every_n_steps: 1
+
+batch_size_finder:
+  _target_: quadra.callbacks.lightning.BatchSizeFinder
+  mode: power
+  steps_per_trial: 3
+  init_val: 2
+  max_trials: 5 # Max 64
+  batch_arg_name: batch_size
+  disable: false
+  find_train_batch_size: true
+  find_validation_batch_size: false
+  find_test_batch_size: false
+  find_predict_batch_size: false

{quadra-2.3.0a1 → quadra-2.3.0a3}/quadra/configs/callbacks/default.yaml

@@ -9,9 +9,6 @@ model_checkpoint:
   filename: "epoch_{epoch:03d}"
   auto_insert_metric_name: False
 
-log_gradients:
-  _target_: quadra.callbacks.mlflow.LogGradients
-  norm: 2
 lr_monitor:
   _target_: pytorch_lightning.callbacks.LearningRateMonitor
   logging_interval: "epoch"

{quadra-2.3.0a1 → quadra-2.3.0a3}/quadra/configs/callbacks/default_anomalib.yaml

@@ -44,9 +44,6 @@ upload_ckpts_as_artifact:
   upload_best_only: true
   delete_after_upload: true
   upload: false
-log_gradients:
-  _target_: quadra.callbacks.mlflow.LogGradients
-  norm: 2
 lr_monitor:
   _target_: pytorch_lightning.callbacks.LearningRateMonitor
   logging_interval: "epoch"

{quadra-2.3.0a1 → quadra-2.3.0a3}/quadra/utils/export.py

@@ -381,7 +381,7 @@ def _safe_export_half_precision_onnx(
             with open(os.devnull, "w") as f, contextlib.redirect_stdout(f):
                 # This function prints a lot of information that is not useful for the user
                 model_fp16 = auto_convert_mixed_precision(
-                    model_fp32, test_data, rtol=0.01, atol=0.001, keep_io_types=False
+                    model_fp32, test_data, rtol=0.01, atol=5e-3, keep_io_types=False
                 )
             onnx.save(model_fp16, export_model_path)