experiment_name: autoencoder_sheet_metal data: dataset: root: data/mvtec_ad_2 category: sheet_metal layout: splits: train: train validation: validation test: test_public good_dir: good bad_dir: bad mask_dir: ground_truth/bad mask_suffix: _mask image_extensions: - .png preprocessing: image_size: - 256 - 1024 crop_size: null mean: - 0.0 - 0.0 - 0.0 std: - 1.0 - 1.0 - 1.0 augmentation: hflip_p: 0.0 vflip_p: 0.0 rotation_degrees: 0.0 brightness: 0.0 contrast: 0.0 blur_sigma: null noise_std: 0.0 seed: 42 model: name: autoencoder base_channels: 32 num_downsamples: 4 latent_channels: 32 training: epochs: 100 batch_size: 4 learning_rate: 0.001 weight_decay: 0.0 optimizer: adam loss: mse grad_accum_steps: 1 num_workers: 0 device: auto scoring: smoothing_sigma: 4.0 image_score: max threshold: image_method: mean_std image_k: 3.0 pixel_quantile: 0.999