From e66a17c86489486f6bbd5099366383e8660923fd Mon Sep 17 00:00:00 2001 From: Joao P C Bertoldo <24547377+jpcbertoldo@users.noreply.github.com> Date: Mon, 21 Nov 2022 10:25:38 +0100 Subject: [PATCH] Fix #699 (#700) * Update lightning_model.py * Update config.yaml * Update config.yaml Co-authored-by: Samet Akcay --- anomalib/models/cflow/config.yaml | 4 ++-- anomalib/models/cflow/lightning_model.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/anomalib/models/cflow/config.yaml b/anomalib/models/cflow/config.yaml index 4337538445..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 @@ -27,7 +26,8 @@ model: condition_vector: 128 coupling_blocks: 8 clamp_alpha: 1.9 - soft_permutation: false + fiber_batch_size: 64 + permute_soft: false lr: 0.0001 early_stopping: patience: 2 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)