Upload 4 files

Browse files

Files changed (4) hide show

config_tree.log +138 -0
epoch_014.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/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

epoch_014.ckpt ADDED Viewed

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

+version https://git-lfs.github.com/spec/v1
+oid sha256:28c652cf1ff0fde8e80df6ca804f285e217488447a00f536ce6872a06d075317
+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,716,,,,,0.9270967841148376,0.9270967841148376,0.2022617608308792
+0,716,,,0.9032673239707947,0.24346967041492462,,,
+1,1433,,,,,0.9294009208679199,0.9294009208679199,0.19298425316810608
+1,1433,,,0.9275213479995728,0.1884833127260208,,,
+2,2150,,,,,0.9351152181625366,0.9351152181625366,0.17905564606189728
+2,2150,,,0.9339120388031006,0.17410194873809814,,,
+3,2867,,,,,0.923778772354126,0.9351152181625366,0.21490544080734253
+3,2867,,,0.938514232635498,0.16192352771759033,,,
+4,3584,,,,,0.9352073669433594,0.9352073669433594,0.1716984510421753
+4,3584,,,0.9415241479873657,0.15334677696228027,,,
+5,4301,,,,,0.9257143139839172,0.9352073669433594,0.18962709605693817
+5,4301,,,0.9461263418197632,0.144466370344162,,,
+6,5018,,,,,0.9268202781677246,0.9352073669433594,0.1882874220609665
+6,5018,,,0.9475658535957336,0.13772191107273102,,,
+7,5735,,,,,0.9397234916687012,0.9397234916687012,0.1684003323316574
+7,5735,,,0.9496815800666809,0.1309342086315155,,,
+8,6452,,,,,0.937972366809845,0.9397234916687012,0.1668853461742401
+8,6452,,,0.9522770643234253,0.12360257655382156,,,
+9,7169,,,,,0.9308755993843079,0.9397234916687012,0.1763351559638977
+9,7169,,,0.9553524851799011,0.117148756980896,,,
+10,7886,,,,,0.9350230693817139,0.9397234916687012,0.16907787322998047
+10,7886,,,0.9570101499557495,0.11205337196588516,,,
+11,8603,,,,,0.936497688293457,0.9397234916687012,0.1680576503276825
+11,8603,,,0.9583624005317688,0.10783351957798004,,,
+12,9320,,,,,0.9333640336990356,0.9397234916687012,0.16861805319786072
+12,9320,,,0.9616995453834534,0.09978552162647247,,,
+13,10037,,,,,0.9367741942405701,0.9397234916687012,0.18279264867305756
+13,10037,,,0.9641205668449402,0.09467099606990814,,,
+14,10754,,,,,0.94036865234375,0.94036865234375,0.17693336308002472
+14,10754,,,0.9655601382255554,0.08994700014591217,,,
+15,11471,,,,,0.932350218296051,0.94036865234375,0.200175479054451
+15,11471,,,0.9680466055870056,0.08448231220245361,,,
+16,12188,,,,,0.9346543550491333,0.94036865234375,0.18900950253009796
+16,12188,,,0.9691807627677917,0.0808669775724411,,,
+17,12905,,,,,0.9276497960090637,0.94036865234375,0.21053890883922577
+17,12905,,,0.9712528586387634,0.07620736956596375,,,
+18,13622,,,,,0.9346543550491333,0.94036865234375,0.20190495252609253
+18,13622,,,0.9711874127388,0.07465444505214691,,,
+19,14339,,,,,0.9354838728904724,0.94036865234375,0.195681631565094
+19,14339,,,0.9733248949050903,0.06832220405340195,,,
+20,14340,0.9349154233932495,0.19932807981967926,,,,,

tags.log ADDED Viewed

	@@ -0,0 +1 @@


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