Upload 4 files

Browse files

Files changed (4) hide show

config_tree.log +138 -0
epoch_017.ckpt +3 -0
metrics.csv +42 -0
tags.log +1 -0

config_tree.log ADDED Viewed

	@@ -0,0 +1,138 @@

+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

epoch_017.ckpt ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:cc7bb2621272abc787f3a40e155b5952137aaacb2732f1203c84f17179258f28
+size 66530862

metrics.csv ADDED Viewed

	@@ -0,0 +1,42 @@

+epoch,step,test/acc,test/loss,train/acc,train/loss,val/acc,val/acc_best,val/loss
+0,597,,,,,0.9710378050804138,0.9710378050804138,0.11186724156141281
+0,597,,,0.9091741442680359,0.2408355176448822,,,
+1,1195,,,,,0.969529390335083,0.9710378050804138,0.12189048528671265
+1,1195,,,0.9560908079147339,0.14466044306755066,,,
+2,1793,,,,,0.9761665463447571,0.9761665463447571,0.09731651842594147
+2,1793,,,0.9602751135826111,0.1332966685295105,,,
+3,2391,,,,,0.976065993309021,0.9761665463447571,0.0973595380783081
+3,2391,,,0.9649563431739807,0.12394490838050842,,,
+4,2989,,,,,0.9776749610900879,0.9776749610900879,0.09066494554281235
+4,2989,,,0.9651917219161987,0.12140671908855438,,,
+5,3587,,,,,0.971540629863739,0.9776749610900879,0.1099431961774826
+5,3587,,,0.9673361778259277,0.11356686800718307,,,
+6,4185,,,,,0.976065993309021,0.9776749610900879,0.09296105802059174
+6,4185,,,0.9674407839775085,0.11387428641319275,,,
+7,4783,,,,,0.9705349802970886,0.9776749610900879,0.10432882606983185
+7,4783,,,0.9676238298416138,0.1122434213757515,,,
+8,5381,,,,,0.9747586250305176,0.9776749610900879,0.09540531784296036
+8,5381,,,0.967754602432251,0.11194324493408203,,,
+9,5979,,,,,0.9788817167282104,0.9788817167282104,0.08850374072790146
+9,5979,,,0.9691406488418579,0.10682261735200882,,,
+10,6577,,,,,0.9770715832710266,0.9788817167282104,0.09269838035106659
+10,6577,,,0.9690883159637451,0.10738109052181244,,,
+11,7175,,,,,0.9772727489471436,0.9788817167282104,0.0941830724477768
+11,7175,,,0.9707097411155701,0.10205083340406418,,,
+12,7773,,,,,0.9787811636924744,0.9788817167282104,0.08852129429578781
+12,7773,,,0.9705790281295776,0.10154617577791214,,,
+13,8371,,,,,0.969529390335083,0.9788817167282104,0.10468596965074539
+13,8371,,,0.9711543321609497,0.10023677349090576,,,
+14,8969,,,,,0.9761665463447571,0.9788817167282104,0.1046816036105156
+14,8969,,,0.970134437084198,0.1029493436217308,,,
+15,9567,,,,,0.9740546941757202,0.9788817167282104,0.10070199519395828
+15,9567,,,0.9706313014030457,0.10002399981021881,,,
+16,10165,,,,,0.9751608967781067,0.9788817167282104,0.1051679402589798
+16,10165,,,0.9723573327064514,0.09687478095293045,,,
+17,10763,,,,,0.9812952280044556,0.9812952280044556,0.08819015324115753
+17,10763,,,0.9710758924484253,0.09853508323431015,,,
+18,11361,,,,,0.9773733019828796,0.9812952280044556,0.0989380031824112
+18,11361,,,0.9729065299034119,0.09388747066259384,,,
+19,11959,,,,,0.9774738550186157,0.9812952280044556,0.09295862913131714
+19,11959,,,0.9722265601158142,0.09592554718255997,,,
+20,11960,0.9825766682624817,0.08600835502147675,,,,,

tags.log ADDED Viewed

	@@ -0,0 +1 @@


1	+ ['pcb', 'deit_tiny_patch16_224.fb_in1k']