| 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: 32 | |
| │ 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: 7.151166416549227e-05 | |
| │ weight_decay: 0.0 | |
| │ scheduler: null | |
| │ net: | |
| │ _target_: src.models.components.vit_rollout_multihead.VitRolloutMultihead | |
| │ model_name: vit_tiny_patch16_224.augreg_in21k_ft_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-14_19-18-34/4/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-14_19-18-34/4 | |
| │ name: csv/ | |
| │ prefix: '' | |
| │ wandb: | |
| │ _target_: lightning.pytorch.loggers.wandb.WandbLogger | |
| │ save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-14_19-18-34/4 | |
| │ offline: false | |
| │ id: null | |
| │ anonymous: null | |
| │ project: DeepVisionXplain | |
| │ log_model: false | |
| │ prefix: '' | |
| │ entity: team_deepvisionxplain | |
| │ group: pcb" | |
| │ tags: | |
| │ - pcb | |
| │ - vit_tiny_patch16_224.augreg_in21k_ft_in1k | |
| │ job_type: '' | |
| │ aim: | |
| │ experiment: vit_tiny_patch16_224.augreg_in21k_ft_in1k_gears | |
| │ | |
| ├── trainer | |
| │ └── _target_: lightning.pytorch.trainer.Trainer | |
| │ default_root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-14_19-18-34/4 | |
| │ 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-14_19-18-34/4 | |
| │ work_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain | |
| │ | |
| ├── extras | |
| │ └── ignore_warnings: false | |
| │ enforce_tags: true | |
| │ print_config: true | |
| │ | |
| ├── task_name | |
| │ └── train | |
| ├── tags | |
| │ └── ['pcb', 'vit_tiny_patch16_224.augreg_in21k_ft_in1k'] | |
| ├── train | |
| │ └── True | |
| ├── test | |
| │ └── True | |
| ├── ckpt_path | |
| │ └── None | |
| ├── seed | |
| │ └── 12345 | |
| └── optimized_metric | |
| └── val/acc_best | |