quadra 0.0.1__py3-none-any.whl → 2.2.7__py3-none-any.whl

quadra/configs/experiment/base/ssl/simclr.yaml

@@ -0,0 +1,48 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: base/ssl
+  - override /backbone: resnet18
+  - override /model: simclr
+  - override /optimizer: lars
+  - override /scheduler: warmup
+  - override /transforms: byol
+  - override /loss: simclr
+  - override /task: ssl
+  - override /trainer: lightning_gpu_fp16
+core:
+  tag: "run"
+  name: "simclr_ssl"
+task:
+  _target_: quadra.tasks.ssl.SimCLR
+
+trainer:
+  devices: [0]
+  max_epochs: 300
+  num_sanity_val_steps: 0
+  check_val_every_n_epoch: 10
+  sync_batchnorm: true
+
+datamodule:
+  num_workers: 16
+  batch_size: 512
+  augmentation_dataset:
+    _target_: quadra.datasets.TwoAugmentationDataset
+    transform:
+      - ${transforms.augmentation1}
+      - ${transforms.augmentation2}
+    dataset: null
+callbacks:
+  model_checkpoint:
+    _target_: pytorch_lightning.callbacks.ModelCheckpoint
+    monitor: "val_acc"
+    mode: "max"
+
+optimizer:
+  lr: 0.0
+  weight_decay: 1e-6
+
+scheduler:
+  init_lr:
+    - 0.3
+  linear_warmup_epochs: 10

quadra/configs/experiment/base/ssl/simsiam.yaml

@@ -0,0 +1,53 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: base/ssl
+  - override /backbone: resnet18
+  - override /model: simsiam
+  - override /optimizer: sgd
+  - override /scheduler: warmup
+  - override /transforms: byol
+  - override /loss: simsiam
+  - override /task: ssl
+  - override /trainer: lightning_gpu_fp16
+core:
+  tag: "run"
+  name: "simsiam_ssl"
+task:
+  _target_: quadra.tasks.ssl.Simsiam
+
+trainer:
+  devices: [0]
+  max_epochs: 300
+  num_sanity_val_steps: 0
+  gradient_clip_val: 2.0
+  check_val_every_n_epoch: 10
+  gradient_clip_algorithm: "value"
+  sync_batchnorm: true
+
+datamodule:
+  num_workers: 16
+  batch_size: 128
+  augmentation_dataset:
+    _target_: quadra.datasets.TwoAugmentationDataset
+    transform:
+      - ${transforms.augmentation1}
+      - ${transforms.augmentation2}
+    dataset: null
+callbacks:
+  model_checkpoint:
+    _target_: pytorch_lightning.callbacks.ModelCheckpoint
+    monitor: "val_acc"
+    mode: "max"
+
+optimizer:
+  lr: 0.0
+
+scheduler:
+  init_lr:
+    - 0.001
+  linear_warmup_epochs: 10
+
+logger:
+  mlflow:
+    experiment_name: ${core.name}

quadra/configs/experiment/custom/cls.yaml

@@ -0,0 +1,12 @@
+# @package _global_
+defaults:
+  - base/classification/classification # extends the default classification configuration
+
+datamodule:
+  data_path: ${oc.env:HOME}/.quadra/datasets/imagenette/imagenette2-320/train/
+
+core:
+  name: custom_custom
+
+model:
+  num_classes: 10

quadra/configs/experiment/default.yaml

@@ -0,0 +1,15 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: null
+  - override /model: null
+  - override /optimizer: null
+  - override /scheduler: null
+  - override /transforms: null
+  - override /loss: null
+  - override /task: default
+  - override /backbone: null
+  - override /trainer: null
+  - override /callbacks: null
+  - override /logger: null
+  - _self_

quadra/configs/experiment/generic/imagenette/classification/default.yaml

@@ -0,0 +1,20 @@
+# @package _global_
+
+defaults:
+  - base/classification/classification
+  - override /datamodule: generic/imagenette/classification/base
+
+core:
+  tag: "run"
+  name: classification_imagenette_${trainer.max_epochs}
+
+trainer:
+  max_epochs: 20
+
+model:
+  num_classes: 10
+
+logger:
+  mlflow:
+    experiment_name: imagenette_classification
+    run_name: ${core.name}

quadra/configs/experiment/generic/imagenette/ssl/barlow.yaml

@@ -0,0 +1,14 @@
+# @package _global_
+
+defaults:
+  - base/ssl/barlow
+  - override /datamodule: generic/imagenette/ssl/base
+
+core:
+  tag: "run"
+  name: "barlow"
+
+logger:
+  mlflow:
+    experiment_name: imagenette_ssl_new
+    run_name: ${core.name}

quadra/configs/experiment/generic/imagenette/ssl/byol.yaml

@@ -0,0 +1,14 @@
+# @package _global_
+
+defaults:
+  - base/ssl/byol
+  - override /datamodule: generic/imagenette/ssl/base
+
+core:
+  tag: "run"
+  name: "byol"
+
+logger:
+  mlflow:
+    experiment_name: imagenette_ssl_new
+    run_name: ${core.name}

quadra/configs/experiment/generic/imagenette/ssl/dino.yaml

@@ -0,0 +1,14 @@
+# @package _global_
+
+defaults:
+  - base/ssl/dino
+  - override /datamodule: generic/imagenette/ssl/base
+
+core:
+  tag: "run"
+  name: "dino"
+
+logger:
+  mlflow:
+    experiment_name: imagenette_ssl_new
+    run_name: ${core.name}

quadra/configs/experiment/generic/imagenette/ssl/simclr.yaml

@@ -0,0 +1,14 @@
+# @package _global_
+
+defaults:
+  - base/ssl/simclr
+  - override /datamodule: generic/imagenette/ssl/base
+
+core:
+  tag: "run"
+  name: "simclr"
+
+logger:
+  mlflow:
+    experiment_name: imagenette_ssl_new
+    run_name: ${core.name}

quadra/configs/experiment/generic/imagenette/ssl/simsiam.yaml

@@ -0,0 +1,14 @@
+# @package _global_
+
+defaults:
+  - base/ssl/simsiam
+  - override /datamodule: generic/imagenette/ssl/base
+
+core:
+  tag: "run"
+  name: "simsiam"
+
+logger:
+  mlflow:
+    experiment_name: imagenette_ssl_new
+    run_name: ${core.name}

quadra/configs/experiment/generic/mnist/anomaly/cfa.yaml

@@ -0,0 +1,34 @@
+# @package _global_
+defaults:
+  - base/anomaly/cfa
+  - override /datamodule: generic/mnist/anomaly/base
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 16
+  test_batch_size: 16
+  task: ${model.dataset.task}
+  good_number: 9
+
+callbacks:
+  score_normalization:
+    threshold_type: "image"
+  early_stopping:
+    patience: 2
+    monitor: validation_image_AUROC
+
+print_config: false
+
+core:
+  tag: "run"
+  test_after_training: true
+  upload_artifacts: true
+  name: cfa_${datamodule.good_number}_${model.model.backbone}_${trainer.max_epochs}
+
+logger:
+  mlflow:
+    experiment_name: mnist-anomaly
+    run_name: ${core.name}
+
+trainer:
+  devices: [0]

quadra/configs/experiment/generic/mnist/anomaly/cflow.yaml

@@ -0,0 +1,33 @@
+# @package _global_
+defaults:
+  - base/anomaly/cflow
+  - override /datamodule: generic/mnist/anomaly/base
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 32
+  task: ${model.dataset.task}
+  good_number: 9
+
+callbacks:
+  score_normalization:
+    threshold_type: "image"
+  early_stopping:
+    patience: 2
+    monitor: validation_image_AUROC
+
+print_config: false
+
+core:
+  tag: "run"
+  test_after_training: true
+  upload_artifacts: true
+  name: cflow_${datamodule.good_number}_${model.model.backbone}_${trainer.max_epochs}
+
+logger:
+  mlflow:
+    experiment_name: mnist-anomaly
+    run_name: ${core.name}
+
+trainer:
+  devices: [0]

quadra/configs/experiment/generic/mnist/anomaly/csflow.yaml

@@ -0,0 +1,33 @@
+# @package _global_
+defaults:
+  - base/anomaly/csflow
+  - override /datamodule: generic/mnist/anomaly/base
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 4
+  task: ${model.dataset.task}
+  good_number: 9
+
+callbacks:
+  score_normalization:
+    threshold_type: "image"
+  early_stopping:
+    patience: 3
+    monitor: validation_image_AUROC
+
+print_config: false
+
+core:
+  tag: "run"
+  test_after_training: true
+  upload_artifacts: true
+  name: csflow_${datamodule.good_number}_efficientnetb5_${trainer.max_epochs}
+
+logger:
+  mlflow:
+    experiment_name: mnist-anomaly
+    run_name: ${core.name}
+
+trainer:
+  devices: [2]

quadra/configs/experiment/generic/mnist/anomaly/draem.yaml

@@ -0,0 +1,33 @@
+# @package _global_
+defaults:
+  - base/anomaly/draem
+  - override /datamodule: generic/mnist/anomaly/base
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 16
+  task: ${model.dataset.task}
+  good_number: 9
+
+callbacks:
+  score_normalization:
+    threshold_type: "image"
+  early_stopping:
+    patience: 20
+    monitor: validation_image_AUROC
+
+print_config: false
+
+core:
+  tag: "run"
+  test_after_training: true
+  upload_artifacts: true
+  name: draem_${datamodule.good_number}_${trainer.max_epochs}
+
+logger:
+  mlflow:
+    experiment_name: mnist-anomaly
+    run_name: ${core.name}
+
+trainer:
+  devices: [0]

quadra/configs/experiment/generic/mnist/anomaly/fastflow.yaml

@@ -0,0 +1,29 @@
+# @package _global_
+defaults:
+  - base/anomaly/fastflow
+  - override /datamodule: generic/mnist/anomaly/base
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 32
+  task: ${model.dataset.task}
+  good_number: 9
+
+callbacks:
+  score_normalization:
+    threshold_type: "image"
+print_config: false
+
+core:
+  tag: "run"
+  test_after_training: true
+  upload_artifacts: true
+  name: fastflow_${datamodule.good_number}_${model.model.backbone}_${trainer.max_epochs}
+
+logger:
+  mlflow:
+    experiment_name: mnist-anomaly
+    run_name: ${core.name}
+
+trainer:
+  devices: [0]

quadra/configs/experiment/generic/mnist/anomaly/inference.yaml

@@ -0,0 +1,27 @@
+# @package _global_
+defaults:
+  - base/anomaly/inference
+  - override /datamodule: generic/mnist/anomaly/base
+
+datamodule:
+  num_workers: 12
+  test_batch_size: 32
+  task: segmentation
+  good_number: 9
+
+core:
+  tag: "run"
+  test_after_training: true
+  upload_artifacts: true
+  name: inference_${datamodule.good_number}
+
+logger:
+  mlflow:
+    experiment_name: mnist-anomaly
+    run_name: ${core.name}
+
+trainer:
+  devices: [0]
+
+task:
+  model_path: ???

quadra/configs/experiment/generic/mnist/anomaly/padim.yaml

@@ -0,0 +1,37 @@
+# @package _global_
+defaults:
+  - base/anomaly/padim
+  - override /datamodule: generic/mnist/anomaly/base
+
+model:
+  model:
+    input_size: [224, 224]
+    backbone: resnet18
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 32
+  task: ${model.dataset.task}
+  good_number: 9
+
+callbacks:
+  score_normalization:
+    threshold_type: "image"
+
+print_config: false
+
+core:
+  tag: "run"
+  test_after_training: true
+  upload_artifacts: true
+  name: padim_${datamodule.good_number}_${model.model.backbone}_${trainer.max_epochs}
+
+logger:
+  mlflow:
+    experiment_name: mnist-anomaly
+    run_name: ${core.name}
+
+trainer:
+  devices: [0]
+  max_epochs: 1
+  check_val_every_n_epoch: ${trainer.max_epochs}

quadra/configs/experiment/generic/mnist/anomaly/patchcore.yaml

@@ -0,0 +1,37 @@
+# @package _global_
+defaults:
+  - base/anomaly/patchcore
+  - override /datamodule: generic/mnist/anomaly/base
+
+model:
+  model:
+    input_size: [224, 224]
+    backbone: resnet18
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 32
+  task: ${model.dataset.task}
+  good_number: 9
+
+callbacks:
+  score_normalization:
+    threshold_type: "image"
+
+print_config: false
+
+core:
+  tag: "run"
+  test_after_training: true
+  upload_artifacts: true
+  name: patchcore_${datamodule.good_number}_${model.model.backbone}_${trainer.max_epochs}
+
+logger:
+  mlflow:
+    experiment_name: mnist-anomaly
+    run_name: ${core.name}
+
+trainer:
+  devices: [0]
+  max_epochs: 1
+  check_val_every_n_epoch: ${trainer.max_epochs}

quadra/configs/experiment/generic/mvtec/anomaly/cfa.yaml

@@ -0,0 +1,34 @@
+# @package _global_
+defaults:
+  - base/anomaly/cfa
+  - override /datamodule: generic/mvtec/anomaly/base
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 16
+  test_batch_size: 16
+  task: ${model.dataset.task}
+  category: bottle
+
+callbacks:
+  score_normalization:
+    threshold_type: "pixel"
+  early_stopping:
+    patience: 2
+    monitor: validation_pixel_AUROC
+
+print_config: false
+
+core:
+  tag: "run"
+  test_after_training: true
+  upload_artifacts: true
+  name: cfa_${datamodule.category}_${model.model.backbone}_${trainer.max_epochs}
+
+logger:
+  mlflow:
+    experiment_name: mvtec-anomaly
+    run_name: ${core.name}
+
+trainer:
+  devices: [2]

quadra/configs/experiment/generic/mvtec/anomaly/cflow.yaml

@@ -0,0 +1,33 @@
+# @package _global_
+defaults:
+  - base/anomaly/cflow
+  - override /datamodule: generic/mvtec/anomaly/base
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 32
+  task: ${model.dataset.task}
+  category: bottle
+
+callbacks:
+  score_normalization:
+    threshold_type: "pixel"
+  early_stopping:
+    patience: 2
+    monitor: validation_pixel_AUROC
+
+print_config: false
+
+core:
+  tag: "run"
+  test_after_training: true
+  upload_artifacts: true
+  name: cflow_${datamodule.category}_${model.model.backbone}_${trainer.max_epochs}
+
+logger:
+  mlflow:
+    experiment_name: mvtec-anomaly
+    run_name: ${core.name}
+
+trainer:
+  devices: [2]

quadra/configs/experiment/generic/mvtec/anomaly/csflow.yaml

@@ -0,0 +1,33 @@
+# @package _global_
+defaults:
+  - base/anomaly/csflow
+  - override /datamodule: generic/mvtec/anomaly/base
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 4
+  task: ${model.dataset.task}
+  category: bottle
+
+callbacks:
+  score_normalization:
+    threshold_type: "pixel"
+  early_stopping:
+    patience: 3
+    monitor: validation_pixel_AUROC
+
+print_config: false
+
+core:
+  tag: "run"
+  test_after_training: true
+  upload_artifacts: true
+  name: csflow_${datamodule.category}_efficientnetb5_${trainer.max_epochs}
+
+logger:
+  mlflow:
+    experiment_name: mvtec-anomaly
+    run_name: ${core.name}
+
+trainer:
+  devices: [2]

quadra/configs/experiment/generic/mvtec/anomaly/draem.yaml

@@ -0,0 +1,33 @@
+# @package _global_
+defaults:
+  - base/anomaly/draem
+  - override /datamodule: generic/mvtec/anomaly/base
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 16
+  task: ${model.dataset.task}
+  category: bottle
+
+callbacks:
+  score_normalization:
+    threshold_type: "pixel"
+  early_stopping:
+    patience: 20
+    monitor: validation_pixel_AUROC
+
+print_config: false
+
+core:
+  tag: "run"
+  test_after_training: true
+  upload_artifacts: true
+  name: draem_${datamodule.category}_${trainer.max_epochs}
+
+logger:
+  mlflow:
+    experiment_name: mvtec-anomaly
+    run_name: ${core.name}
+
+trainer:
+  devices: [2]

quadra/configs/experiment/generic/mvtec/anomaly/efficient_ad.yaml

@@ -0,0 +1,38 @@
+# @package _global_
+defaults:
+  - base/anomaly/efficient_ad
+  - override /datamodule: generic/mvtec/anomaly/base
+
+transforms:
+  input_height: 256
+  input_width: 256
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 1
+  test_batch_size: 1
+  task: ${model.dataset.task}
+  category: hazelnut
+
+callbacks:
+  score_normalization:
+    threshold_type: "pixel"
+
+print_config: false
+
+core:
+  tag: "run"
+  test_after_training: true
+  upload_artifacts: true
+  name: efficient_ad_${datamodule.category}_${model.model.model_size}_${trainer.max_epochs}
+
+logger:
+  mlflow:
+    experiment_name: mvtec-anomaly
+    run_name: ${core.name}
+
+trainer:
+  devices: [0]
+  max_epochs: 50
+  max_steps: 20000
+  check_val_every_n_epoch: ${trainer.max_epochs}