refactor: Remove gradient logging from default callback, add automatic batch size callback to all callback

Author: lorenzomammana

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,10 +9,6 @@ model_checkpoint:
   filename: "epoch_{epoch:03d}"
   auto_insert_metric_name: False
 
-log_gradients:
-  _target_: quadra.callbacks.mlflow.LogGradients
-  norm: 2
-  disable: True
 lr_monitor:
   _target_: pytorch_lightning.callbacks.LearningRateMonitor
   logging_interval: "epoch"

quadra/configs/callbacks/default_anomalib.yaml

@@ -44,10 +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
-  disable: true
 lr_monitor:
   _target_: pytorch_lightning.callbacks.LearningRateMonitor
   logging_interval: "epoch"