CONFIG ├── data │ └── _target_: src.data.dir_datamodule.DirDataModule │ data_dir: /media/gpu0/data/lukas/magistras/data/gears/ │ train_subdir: train/ │ val_subdir: val/ │ test_subdir: test/ │ image_subdir: images/ │ label_subdir: labels/ │ batch_size: 64 │ num_workers: 10 │ pin_memory: false │ preprocessor: │ _target_: src.data.components.tile_processor.TilingProcessor │ tile_width: 224 │ tile_height: 224 │ min_defective_area: 0.01 │ overlap: 2 │ step_size: 10 │ iterate_over_defective_areas: false │ save_every_second_good_tile: true │ ├── model │ └── _target_: src.models.training_module.TrainingLitModule │ optimizer: │ _target_: torch.optim.Adam │ _partial_: true │ lr: 0.0028369961259166574 │ weight_decay: 0.0 │ scheduler: │ _target_: torch.optim.lr_scheduler.ReduceLROnPlateau │ _partial_: true │ mode: min │ factor: 0.1 │ patience: 10 │ net: │ _target_: src.models.components.cnn_cam_multihead.CNNCAMMultihead │ backbone: mobilenet_v3_large │ multi_head: false │ return_nodes: │ features.13.block.0: layerout │ weights: IMAGENET1K_V1 │ compile: false │ ├── callbacks │ └── model_checkpoint: │ _target_: lightning.pytorch.callbacks.ModelCheckpoint │ dirpath: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-13_01-01-07/2/checkpoints │ filename: epoch_{epoch:03d} │ monitor: val/acc │ verbose: false │ save_last: true │ save_top_k: 1 │ mode: max │ auto_insert_metric_name: false │ save_weights_only: false │ every_n_train_steps: null │ train_time_interval: null │ every_n_epochs: null │ save_on_train_epoch_end: null │ early_stopping: │ _target_: lightning.pytorch.callbacks.EarlyStopping │ monitor: val/acc │ min_delta: 0.0 │ patience: 100 │ verbose: false │ mode: max │ strict: true │ check_finite: true │ stopping_threshold: null │ divergence_threshold: null │ check_on_train_epoch_end: null │ model_summary: │ _target_: lightning.pytorch.callbacks.RichModelSummary │ max_depth: -1 │ rich_progress_bar: │ _target_: lightning.pytorch.callbacks.RichProgressBar │ ├── logger │ └── csv: │ _target_: lightning.pytorch.loggers.csv_logs.CSVLogger │ save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-13_01-01-07/2 │ name: csv/ │ prefix: '' │ wandb: │ _target_: lightning.pytorch.loggers.wandb.WandbLogger │ save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-13_01-01-07/2 │ offline: false │ id: null │ anonymous: null │ project: DeepVisionXplain │ log_model: false │ prefix: '' │ entity: team_deepvisionxplain │ group: gears │ tags: │ - gears │ - mobilenet_v3_large │ - downscaled │ job_type: '' │ aim: │ experiment: mobilenet_v3_large_full_gears │ ├── trainer │ └── _target_: lightning.pytorch.trainer.Trainer │ default_root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-13_01-01-07/2 │ min_epochs: 10 │ max_epochs: 20 │ accelerator: gpu │ devices: 1 │ check_val_every_n_epoch: 1 │ deterministic: false │ ├── paths │ └── root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain │ data_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/data/ │ log_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/ │ output_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-13_01-01-07/2 │ work_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain │ ├── extras │ └── ignore_warnings: false │ enforce_tags: true │ print_config: true │ ├── task_name │ └── train ├── tags │ └── ['gears', 'mobilenet_v3_large', 'downscaled'] ├── train │ └── True ├── test │ └── True ├── ckpt_path │ └── None ├── seed │ └── 12345 └── optimized_metric └── val/acc_best