DeepVisionXplain/deit_tiny_patch16_224.fb_in1k_pcb
0
1CONFIG2├── data3│ └── _target_: src.data.dir_datamodule.DirDataModule 4│ data_dir: /media/gpu0/data/lukas/magistras/data/pcb/ 5│ train_subdir: train/ 6│ val_subdir: val/ 7│ test_subdir: test/ 8│ image_subdir: images/ 9│ label_subdir: labels/ 10│ batch_size: 64 11│ num_workers: 10 12│ pin_memory: false 13│ preprocessor: 14│ _target_: src.data.components.tile_processor.TilingProcessor 15│ tile_width: 224 16│ tile_height: 224 17│ min_defective_area: 0.01 18│ overlap: 2 19│ step_size: 10 20│ iterate_over_defective_areas: false 21│ save_every_second_good_tile: true 22│ 23├── model24│ └── _target_: src.models.training_module.TrainingLitModule 25│ optimizer: 26│ _target_: torch.optim.AdamW 27│ _partial_: true 28│ lr: 8.09857128646671e-05 29│ weight_decay: 0.0 30│ scheduler: null 31│ net: 32│ _target_: src.models.components.vit_rollout_multihead.VitRolloutMultihead 33│ model_name: deit_tiny_patch16_224.fb_in1k 34│ pretrained: true 35│ weight_path: null 36│ output_size: 1 37│ return_nodes: attn_drop 38│ head_name: head 39│ img_size: 224 40│ discard_ratio: 0.2 41│ head_fusion: mean 42│ visualize: false 43│ compile: false 44│ 45├── callbacks46│ └── model_checkpoint: 47│ _target_: lightning.pytorch.callbacks.ModelCheckpoint 48│ dirpath: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-17_23-37-48/9/checkpoints 49│ filename: epoch_{epoch:03d} 50│ monitor: val/acc 51│ verbose: false 52│ save_last: true 53│ save_top_k: 1 54│ mode: max 55│ auto_insert_metric_name: false 56│ save_weights_only: false 57│ every_n_train_steps: null 58│ train_time_interval: null 59│ every_n_epochs: null 60│ save_on_train_epoch_end: null 61│ early_stopping: 62│ _target_: lightning.pytorch.callbacks.EarlyStopping 63│ monitor: val/acc 64│ min_delta: 0.0 65│ patience: 100 66│ verbose: false 67│ mode: max 68│ strict: true 69│ check_finite: true 70│ stopping_threshold: null 71│ divergence_threshold: null 72│ check_on_train_epoch_end: null 73│ model_summary: 74│ _target_: lightning.pytorch.callbacks.RichModelSummary 75│ max_depth: -1 76│ rich_progress_bar: 77│ _target_: lightning.pytorch.callbacks.RichProgressBar 78│ 79├── logger80│ └── csv: 81│ _target_: lightning.pytorch.loggers.csv_logs.CSVLogger 82│ save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-17_23-37-48/9 83│ name: csv/ 84│ prefix: '' 85│ wandb: 86│ _target_: lightning.pytorch.loggers.wandb.WandbLogger 87│ save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-17_23-37-48/9 88│ offline: false 89│ id: null 90│ anonymous: null 91│ project: DeepVisionXplain 92│ log_model: false 93│ prefix: '' 94│ entity: team_deepvisionxplain 95│ group: pcb" 96│ tags: 97│ - pcb 98│ - deit_tiny_patch16_224.fb_in1k 99│ job_type: '' 100│ aim: 101│ experiment: deit_tiny_patch16_224.fb_in1k_pcb 102│ 103├── trainer104│ └── _target_: lightning.pytorch.trainer.Trainer 105│ default_root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-17_23-37-48/9 106│ min_epochs: 10 107│ max_epochs: 20 108│ accelerator: gpu 109│ devices: 1 110│ check_val_every_n_epoch: 1 111│ deterministic: false 112│ 113├── paths114│ └── root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain 115│ data_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/data/ 116│ log_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/ 117│ output_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-17_23-37-48/9 118│ work_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain 119│ 120├── extras121│ └── ignore_warnings: false 122│ enforce_tags: true 123│ print_config: true 124│ 125├── task_name126│ └── train 127├── tags128│ └── ['pcb', 'deit_tiny_patch16_224.fb_in1k'] 129├── train130│ └── True 131├── test132│ └── True 133├── ckpt_path134│ └── None 135├── seed136│ └── 12345 137└── optimized_metric138 └── val/acc_best 139