| 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: 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.AdamW |
| │ _partial_: true |
| │ lr: 8.09857128646671e-05 |
| │ weight_decay: 0.0 |
| │ scheduler: null |
| │ net: |
| │ _target_: src.models.components.vit_rollout_multihead.VitRolloutMultihead |
| │ model_name: deit_tiny_patch16_224.fb_in1k |
| │ pretrained: true |
| │ weight_path: null |
| │ output_size: 1 |
| │ return_nodes: attn_drop |
| │ head_name: head |
| │ img_size: 224 |
| │ discard_ratio: 0.2 |
| │ head_fusion: mean |
| │ visualize: false |
| │ compile: false |
| │ |
| ├── callbacks |
| │ └── model_checkpoint: |
| │ _target_: lightning.pytorch.callbacks.ModelCheckpoint |
| │ dirpath: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-17_23-37-48/9/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-17_23-37-48/9 |
| │ name: csv/ |
| │ prefix: '' |
| │ wandb: |
| │ _target_: lightning.pytorch.loggers.wandb.WandbLogger |
| │ save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-17_23-37-48/9 |
| │ offline: false |
| │ id: null |
| │ anonymous: null |
| │ project: DeepVisionXplain |
| │ log_model: false |
| │ prefix: '' |
| │ entity: team_deepvisionxplain |
| │ group: pcb" |
| │ tags: |
| │ - pcb |
| │ - deit_tiny_patch16_224.fb_in1k |
| │ job_type: '' |
| │ aim: |
| │ experiment: deit_tiny_patch16_224.fb_in1k_pcb |
| │ |
| ├── trainer |
| │ └── _target_: lightning.pytorch.trainer.Trainer |
| │ default_root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-17_23-37-48/9 |
| │ 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-17_23-37-48/9 |
| │ work_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain |
| │ |
| ├── extras |
| │ └── ignore_warnings: false |
| │ enforce_tags: true |
| │ print_config: true |
| │ |
| ├── task_name |
| │ └── train |
| ├── tags |
| │ └── ['pcb', 'deit_tiny_patch16_224.fb_in1k'] |
| ├── train |
| │ └── True |
| ├── test |
| │ └── True |
| ├── ckpt_path |
| │ └── None |
| ├── seed |
| │ └── 12345 |
| └── optimized_metric |
| └── val/acc_best |
|
|
