From caf4e4ea1cd136ccd75ecd0ad646ebdb97679daa Mon Sep 17 00:00:00 2001
From: Joao P C Bertoldo <24547377+jpcbertoldo@users.noreply.github.com>
Date: Tue, 8 Nov 2022 15:42:25 +0100
Subject: [PATCH 1/3] Update lightning_model.py

---
 anomalib/models/cflow/lightning_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/anomalib/models/cflow/lightning_model.py b/anomalib/models/cflow/lightning_model.py
index 3065d4c506..9ac66cad7a 100644
--- a/anomalib/models/cflow/lightning_model.py
+++ b/anomalib/models/cflow/lightning_model.py
@@ -186,12 +186,12 @@ def __init__(self, hparams: Union[DictConfig, ListConfig]) -> None:
             backbone=hparams.model.backbone,
             layers=hparams.model.layers,
             pre_trained=hparams.model.pre_trained,
-            fiber_batch_size=hparams.dataset.fiber_batch_size,
+            fiber_batch_size=hparams.model.fiber_batch_size,
             decoder=hparams.model.decoder,
             condition_vector=hparams.model.condition_vector,
             coupling_blocks=hparams.model.coupling_blocks,
             clamp_alpha=hparams.model.clamp_alpha,
-            permute_soft=hparams.model.soft_permutation,
+            permute_soft=hparams.model.permute_soft,
         )
         self.hparams: Union[DictConfig, ListConfig]  # type: ignore
         self.save_hyperparameters(hparams)

From 20f27a7a0a44697f4ee93f3dadcefe470303f233 Mon Sep 17 00:00:00 2001
From: Joao P C Bertoldo <24547377+jpcbertoldo@users.noreply.github.com>
Date: Tue, 8 Nov 2022 15:45:10 +0100
Subject: [PATCH 2/3] Update config.yaml

---
 anomalib/models/cflow/config.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/anomalib/models/cflow/config.yaml b/anomalib/models/cflow/config.yaml
index 4337538445..7a3a72dbed 100644
--- a/anomalib/models/cflow/config.yaml
+++ b/anomalib/models/cflow/config.yaml
@@ -27,6 +27,7 @@ model:
   condition_vector: 128
   coupling_blocks: 8
   clamp_alpha: 1.9
+  fiber_batch_size: 64
   soft_permutation: false
   lr: 0.0001
   early_stopping:

From 5bae657436be6f2b03e952a9b89756cd0d8b6493 Mon Sep 17 00:00:00 2001
From: Joao P C Bertoldo <24547377+jpcbertoldo@users.noreply.github.com>
Date: Tue, 8 Nov 2022 16:49:05 +0100
Subject: [PATCH 3/3] Update config.yaml

---
 anomalib/models/cflow/config.yaml | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/anomalib/models/cflow/config.yaml b/anomalib/models/cflow/config.yaml
index 7a3a72dbed..ea2295ae29 100644
--- a/anomalib/models/cflow/config.yaml
+++ b/anomalib/models/cflow/config.yaml
@@ -8,7 +8,6 @@ dataset:
   train_batch_size: 16
   test_batch_size: 16
   inference_batch_size: 16
-  fiber_batch_size: 64
   num_workers: 8
   transform_config:
     train: null
@@ -28,7 +27,7 @@ model:
   coupling_blocks: 8
   clamp_alpha: 1.9
   fiber_batch_size: 64
-  soft_permutation: false
+  permute_soft: false
   lr: 0.0001
   early_stopping:
     patience: 2