DeepVisionXplain
/

vit_tiny_patch16_224.augreg_in21k_ft_in1k_pcb

Model card Files Files and versions Community

lus105 commited on Mar 15, 2024

Commit

8aa60cf

verified ·

1 Parent(s): 7c8a7dd

Upload 4 files

Browse files

Files changed (4) hide show

config_tree.log +138 -0
epoch_018.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: 32
+│       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: 7.151166416549227e-05
+│         weight_decay: 0.0
+│       scheduler: null
+│       net:
+│         _target_: src.models.components.vit_rollout_multihead.VitRolloutMultihead
+│         model_name: vit_tiny_patch16_224.augreg_in21k_ft_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-14_19-18-34/4/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-14_19-18-34/4
+│         name: csv/
+│         prefix: ''
+│       wandb:
+│         _target_: lightning.pytorch.loggers.wandb.WandbLogger
+│         save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-14_19-18-34/4
+│         offline: false
+│         id: null
+│         anonymous: null
+│         project: DeepVisionXplain
+│         log_model: false
+│         prefix: ''
+│         entity: team_deepvisionxplain
+│         group: pcb"
+│         tags:
+│         - pcb
+│         - vit_tiny_patch16_224.augreg_in21k_ft_in1k
+│         job_type: ''
+│       aim:
+│         experiment: vit_tiny_patch16_224.augreg_in21k_ft_in1k_gears
+│
+├── trainer
+│   └── _target_: lightning.pytorch.trainer.Trainer
+│       default_root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-14_19-18-34/4
+│       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-14_19-18-34/4
+│       work_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain
+│
+├── extras
+│   └── ignore_warnings: false
+│       enforce_tags: true
+│       print_config: true
+│
+├── task_name
+│   └── train
+├── tags
+│   └── ['pcb', 'vit_tiny_patch16_224.augreg_in21k_ft_in1k']
+├── train
+│   └── True
+├── test
+│   └── True
+├── ckpt_path
+│   └── None
+├── seed
+│   └── 12345
+└── optimized_metric
+    └── val/acc_best

epoch_018.ckpt ADDED Viewed

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

+version https://git-lfs.github.com/spec/v1
+oid sha256:502aef680c2106b8d82226cd0ec8ffa3300eb4feb743848c2a5e03bec267c247
+size 66530926

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,1194,,,,,0.973551869392395,0.973551869392395,0.09961999952793121
+0,1194,,,0.9400073289871216,0.1793513149023056,,,
+1,2389,,,,,0.9773733019828796,0.9773733019828796,0.08894068002700806
+1,2389,,,0.9636225700378418,0.12740583717823029,,,
+2,3584,,,,,0.9761665463447571,0.9773733019828796,0.09505005925893784
+2,3584,,,0.9659762382507324,0.11955148726701736,,,
+3,4779,,,,,0.9764682054519653,0.9773733019828796,0.1030309870839119
+3,4779,,,0.9671007990837097,0.116389200091362,,,
+4,5974,,,,,0.977775514125824,0.977775514125824,0.08794596046209335
+4,5974,,,0.9681206941604614,0.11324760317802429,,,
+5,7169,,,,,0.973753035068512,0.977775514125824,0.10174654424190521
+5,7169,,,0.9688791036605835,0.10822844505310059,,,
+6,8364,,,,,0.9769710302352905,0.977775514125824,0.0967702567577362
+6,8364,,,0.9697944521903992,0.10680365562438965,,,
+7,9559,,,,,0.9779766798019409,0.9779766798019409,0.09349807351827621
+7,9559,,,0.9699774980545044,0.1049199104309082,,,
+8,10754,,,,,0.9750603437423706,0.9779766798019409,0.09106110036373138
+8,10754,,,0.9705005288124084,0.10296483337879181,,,
+9,11949,,,,,0.9768704771995544,0.9779766798019409,0.08709587901830673
+9,11949,,,0.9714420437812805,0.10124092549085617,,,
+10,13144,,,,,0.9762670993804932,0.9779766798019409,0.0949004516005516
+10,13144,,,0.9714681506156921,0.09946727752685547,,,
+11,14339,,,,,0.9796862602233887,0.9796862602233887,0.08996349573135376
+11,14339,,,0.972043514251709,0.09838294237852097,,,
+12,15534,,,,,0.9756637215614319,0.9796862602233887,0.09498219937086105
+12,15534,,,0.972749650478363,0.09623654931783676,,,
+13,16729,,,,,0.9795857071876526,0.9796862602233887,0.08524612337350845
+13,16729,,,0.972566545009613,0.09618011116981506,,,
+14,17924,,,,,0.9756637215614319,0.9796862602233887,0.09883344918489456
+14,17924,,,0.9731941819190979,0.0932031199336052,,,
+15,19119,,,,,0.9789823293685913,0.9796862602233887,0.09219285100698471
+15,19119,,,0.972749650478363,0.09415934234857559,,,
+16,20314,,,,,0.9787811636924744,0.9796862602233887,0.08861564099788666
+16,20314,,,0.9732465147972107,0.09321973472833633,,,
+17,21509,,,,,0.9773733019828796,0.9796862602233887,0.09647410362958908
+17,21509,,,0.973900318145752,0.09043314307928085,,,
+18,22704,,,,,0.9830048084259033,0.9830048084259033,0.08431670069694519
+18,22704,,,0.9736911058425903,0.09099637717008591,,,
+19,23899,,,,,0.9799879193305969,0.9830048084259033,0.09198801219463348
+19,23899,,,0.9735603332519531,0.09155704081058502,,,
+20,23900,0.9836220741271973,0.08370564132928848,,,,,

tags.log ADDED Viewed

	@@ -0,0 +1 @@


1	+ ['pcb', 'vit_tiny_patch16_224.augreg_in21k_ft_in1k']