File size: 23,583 Bytes
de8a97c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 |
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
|