quadra 2.3.0a1__py3-none-any.whl → 2.3.0a3__py3-none-any.whl

quadra/__init__.py

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

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/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/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/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)
 

{quadra-2.3.0a1.dist-info → quadra-2.3.0a3.dist-info}/METADATA

@@ -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.dist-info → quadra-2.3.0a3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-quadra/__init__.py,sha256=vxRo6uJuoohmIxvYi6McQA_GMN9wnOLDPrQ1Ub7Crsg,114
+quadra/__init__.py,sha256=Y2WIePdnaKPseYL22K9Grnpqwa7aMazgU0CWRWBoaJc,114
 quadra/callbacks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 quadra/callbacks/anomalib.py,sha256=Z0Xx3M9UEMYGY0BpSsqemgQSQ6jAr4SrvlbDg_oG9C8,11913
 quadra/callbacks/lightning.py,sha256=1OTM6fB7qBVLPWNBAZJOb3B00q0kAxMWkPjTEn9YgF0,20182
@@ -32,9 +32,9 @@ quadra/configs/backbone/vit16_base.yaml,sha256=LWMAx3uwKQClDds1zUMzdNercHgO3kanV
 quadra/configs/backbone/vit16_small.yaml,sha256=8VkdiFCDugoSwz3tEbTNYp54P2eHElC8dR9Tea2Qir8,198
 quadra/configs/backbone/vit16_tiny.yaml,sha256=iHk8eUeieg-MJF4YA7m7p5WiTz5KT_rmNmYA5LLjdaw,197
 quadra/configs/backbone/xcit_tiny_24_p8_224.yaml,sha256=GnlQpUcahn_kZpj109kCC_5X4HP9-BSlgXRkiQJTlGU,196
-quadra/configs/callbacks/all.yaml,sha256=KYEDjqAf7lC5gpU8uUjlfpETIiDaWq5AlnQN-OCOTZU,1113
-quadra/configs/callbacks/default.yaml,sha256=sJBXHcrgycjGiz2lfSn54s1c1fbuED2QLKMaJdos078,1254
-quadra/configs/callbacks/default_anomalib.yaml,sha256=spzxnmnl5RRWOxzWSIZDmtb3cz9bGAErZEO-PIAFoqE,2330
+quadra/configs/callbacks/all.yaml,sha256=LZx8d0apwv9t0KlKqzFCrYo2NpuNBgcUnHhG_kud5fQ,1437
+quadra/configs/callbacks/default.yaml,sha256=ZFPU1bm36hJsxI-85uiJx7TpX7qWkR8dibBKWtES4Yc,1180
+quadra/configs/callbacks/default_anomalib.yaml,sha256=FjjSj6HgMvH18MV4AKR4Ew0pOerPefj9eUGinUkODLE,2256
 quadra/configs/config.yaml,sha256=IULhqUF8Z7Cqr5Xx41EGj8dwtPQWSRiZ5jwzNg0Rjwk,686
 quadra/configs/core/default.yaml,sha256=IfgjKHXuOknq3CKvKKBqMPfiqmZSUOnc80Q2Jkbm7go,239
 quadra/configs/datamodule/base/anomaly.yaml,sha256=CILLAoQHunrT4BN0ynOzizTX29k-7B9vDVIZUYm-cBU,377
@@ -262,7 +262,7 @@ quadra/utils/anomaly.py,sha256=49vFvT5-4SxczsEM2Akcut_M1DDwKlOVdGv36oLTgR0,4067
 quadra/utils/classification.py,sha256=7Phbywpd1CNzwLM5F7zV9FrzdHGd6EkR6LRTRpkerTc,24882
 quadra/utils/deprecation.py,sha256=zF_S-yqenaZxRBOudhXts0mX763WjEUWCnHd09TZnwY,852
 quadra/utils/evaluation.py,sha256=5Occ6ONsjnvMkaFFjohtgSvffxEKfjdvMbxqzgtmiHQ,18538
-quadra/utils/export.py,sha256=PoKaADeSwGPIW5TEHP3Klw_SUSV1gP1dIN4CN-AgDm0,20836
+quadra/utils/export.py,sha256=2yRA9KQgZGi3742JyDTfDSdmDJCOiv5pCObp2QoA330,20835
 quadra/utils/imaging.py,sha256=Cz7sGb_axEmnGcwQJP2djFZpIpGCPFIBGT8NWVV-OOE,866
 quadra/utils/logger.py,sha256=tQJ4xpTAFKx1g-UUm5K1x7zgoP6qoXpcUHQyu0rOr1w,556
 quadra/utils/mlflow.py,sha256=7E09JShGm2qO7bLb_8srA_RYdVAudxeDYmX7pMdjoVU,3524
@@ -293,8 +293,8 @@ quadra/utils/validator.py,sha256=eFCGr0ss1gYSpsL31JbsCXPZUMJAI9_H-mGodt6UGsU,466
 quadra/utils/visualization.py,sha256=UvGHX0dumfjpT_KX3Yc1W2B5sAXXaZZWIwXQAi4sdoQ,15950
 quadra/utils/vit_explainability.py,sha256=hY0awehj6UkyBhnBlW5uWoJTsBfgow5Nll9fAqrzmMo,13337
 hydra_plugins/quadra_searchpath_plugin.py,sha256=AAn4TzR87zUK7nwSsK-KoqALiPtfQ8FvX3fgZPTGIJ0,1189
-quadra-2.3.0a1.dist-info/LICENSE,sha256=8cTbQtcWa02YJoSpMeV_gxj3jpMTkxvl-w3WJ5gV_QE,11342
-quadra-2.3.0a1.dist-info/METADATA,sha256=zK3qs3yIokFgVMZzZ2jfV23i-1hBFvrFtBMxmKuZM0w,17625
-quadra-2.3.0a1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-quadra-2.3.0a1.dist-info/entry_points.txt,sha256=sRYonBZyx-sAJeWcQNQoVQIU5lm02cnCQt6b15k0WHU,43
-quadra-2.3.0a1.dist-info/RECORD,,
+quadra-2.3.0a3.dist-info/LICENSE,sha256=8cTbQtcWa02YJoSpMeV_gxj3jpMTkxvl-w3WJ5gV_QE,11342
+quadra-2.3.0a3.dist-info/METADATA,sha256=CKo1FLWFEb14VrBJ-ClCXHWrSHdgBAqzo3tosK2qdeo,17600
+quadra-2.3.0a3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+quadra-2.3.0a3.dist-info/entry_points.txt,sha256=sRYonBZyx-sAJeWcQNQoVQIU5lm02cnCQt6b15k0WHU,43
+quadra-2.3.0a3.dist-info/RECORD,,