| CONFIG | |
| ├── data | |
| │ └── _target_: src.data.dir_datamodule.DirDataModule | |
| │ data_dir: /media/gpu0/data/lukas/magistras/data/pcb/ | |
| │ 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: 10 | |
| │ step_size: 10 | |
| │ iterate_over_defective_areas: true | |
| │ 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: efficientnet_v2_s | |
| │ multi_head: false | |
| │ return_nodes: | |
| │ features.7: 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-12_07-27-45/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-12_07-27-45/2 | |
| │ name: csv/ | |
| │ prefix: '' | |
| │ wandb: | |
| │ _target_: lightning.pytorch.loggers.wandb.WandbLogger | |
| │ save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-12_07-27-45/2 | |
| │ offline: false | |
| │ id: null | |
| │ anonymous: null | |
| │ project: DeepVisionXplain | |
| │ log_model: false | |
| │ prefix: '' | |
| │ entity: team_deepvisionxplain | |
| │ group: pcb" | |
| │ tags: | |
| │ - pcb | |
| │ - efficientnet_v2_s | |
| │ job_type: '' | |
| │ aim: | |
| │ experiment: efficientnet_v2_s_pcb | |
| │ | |
| ├── trainer | |
| │ └── _target_: lightning.pytorch.trainer.Trainer | |
| │ default_root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-12_07-27-45/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-12_07-27-45/2 | |
| │ work_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain | |
| │ | |
| ├── extras | |
| │ └── ignore_warnings: false | |
| │ enforce_tags: true | |
| │ print_config: true | |
| │ | |
| ├── task_name | |
| │ └── train | |
| ├── tags | |
| │ └── ['pcb', 'efficientnet_v2_s'] | |
| ├── train | |
| │ └── True | |
| ├── test | |
| │ └── True | |
| ├── ckpt_path | |
| │ └── None | |
| ├── seed | |
| │ └── 12345 | |
| └── optimized_metric | |
| └── val/acc_best | |