quadra 2.2.7__py3-none-any.whl → 2.3.0a1__py3-none-any.whl

quadra/__init__.py

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

quadra/configs/callbacks/all.yaml

@@ -30,16 +30,3 @@ 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,6 +9,9 @@ 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,6 +44,9 @@ 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.2.7.dist-info → quadra-2.3.0a1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quadra
-Version: 2.2.7
+Version: 2.3.0a1
 Summary: Deep Learning experiment orchestration library
 Home-page: https://orobix.github.io/quadra
 License: Apache-2.0
@@ -35,12 +35,12 @@ Requires-Dist: mlflow-skinny (>=2.3.1,<3.0.0)
 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"
-Requires-Dist: onnxruntime_gpu (==1.17.0) ; extra == "onnx"
+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: pydantic (>=1.10.10)
 Requires-Dist: python_dotenv (>=0.21,<0.22)
 Requires-Dist: pytorch_lightning (>=2.1,<2.2)
 Requires-Dist: rich (>=13.2,<13.3)

{quadra-2.2.7.dist-info → quadra-2.3.0a1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-quadra/__init__.py,sha256=b8bA8-5sXhY8EPisrxKFTKQoKLsZ3NLcstyR_r0ZOjg,112
+quadra/__init__.py,sha256=vxRo6uJuoohmIxvYi6McQA_GMN9wnOLDPrQ1Ub7Crsg,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=LZx8d0apwv9t0KlKqzFCrYo2NpuNBgcUnHhG_kud5fQ,1437
-quadra/configs/callbacks/default.yaml,sha256=ZFPU1bm36hJsxI-85uiJx7TpX7qWkR8dibBKWtES4Yc,1180
-quadra/configs/callbacks/default_anomalib.yaml,sha256=FjjSj6HgMvH18MV4AKR4Ew0pOerPefj9eUGinUkODLE,2256
+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/config.yaml,sha256=IULhqUF8Z7Cqr5Xx41EGj8dwtPQWSRiZ5jwzNg0Rjwk,686
 quadra/configs/core/default.yaml,sha256=IfgjKHXuOknq3CKvKKBqMPfiqmZSUOnc80Q2Jkbm7go,239
 quadra/configs/datamodule/base/anomaly.yaml,sha256=CILLAoQHunrT4BN0ynOzizTX29k-7B9vDVIZUYm-cBU,377
@@ -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.2.7.dist-info/LICENSE,sha256=8cTbQtcWa02YJoSpMeV_gxj3jpMTkxvl-w3WJ5gV_QE,11342
-quadra-2.2.7.dist-info/METADATA,sha256=Sd9YRCYV7zmk0OrLSay9D6dvtq-t-fITBqD5ixrngI4,17623
-quadra-2.2.7.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-quadra-2.2.7.dist-info/entry_points.txt,sha256=sRYonBZyx-sAJeWcQNQoVQIU5lm02cnCQt6b15k0WHU,43
-quadra-2.2.7.dist-info/RECORD,,
+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.2.7.dist-info → quadra-2.3.0a1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.1
+Generator: poetry-core 1.9.0
 Root-Is-Purelib: true
 Tag: py3-none-any