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.AdamW                                                                                                                                                     
│         _partial_: true                                                                                                                                                                 
│         lr: 6.538565605901445e-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-16_16-40-40/7/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-16_16-40-40/7                                                                          
│         name: csv/                                                                                                                                                                      
│         prefix: ''                                                                                                                                                                      
│       wandb:                                                                                                                                                                            
│         _target_: lightning.pytorch.loggers.wandb.WandbLogger                                                                                                                           
│         save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-16_16-40-40/7                                                                          
│         offline: false                                                                                                                                                                  
│         id: null                                                                                                                                                                        
│         anonymous: null                                                                                                                                                                 
│         project: DeepVisionXplain                                                                                                                                                       
│         log_model: false                                                                                                                                                                
│         prefix: ''                                                                                                                                                                      
│         entity: team_deepvisionxplain                                                                                                                                                   
│         group: gears                                                                                                                                                                    
│         tags:                                                                                                                                                                           
│         - gears                                                                                                                                                                         
│         - deit_tiny_patch16_224.fb_in1k                                                                                                                                                 
│         job_type: ''                                                                                                                                                                    
│       aim:                                                                                                                                                                              
│         experiment: deit_tiny_patch16_224.fb_in1k_gears                                                                                                                                 
│                                                                                                                                                                                         
├── trainer
│   └── _target_: lightning.pytorch.trainer.Trainer                                                                                                                                       
│       default_root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-16_16-40-40/7                                                                    
│       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-16_16-40-40/7                                                                          
│       work_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain                                                                                                                       
│                                                                                                                                                                                         
├── extras
│   └── ignore_warnings: false                                                                                                                                                            
│       enforce_tags: true                                                                                                                                                                
│       print_config: true                                                                                                                                                                
│                                                                                                                                                                                         
├── task_name
│   └── train                                                                                                                                                                             
├── tags
│   └── ['gears', 'deit_tiny_patch16_224.fb_in1k']                                                                                                                                        
├── train
│   └── True                                                                                                                                                                              
├── test
│   └── True                                                                                                                                                                              
├── ckpt_path
│   └── None                                                                                                                                                                              
├── seed
│   └── 12345                                                                                                                                                                             
└── optimized_metric
    └── val/acc_best