DeepVisionXplain
/

efficientnet_v2_s_downscaled_pcb

Model card Files Files and versions Community

lus105 commited on Mar 12, 2024

Commit

d2112d2

verified ·

1 Parent(s): e284dde

Upload 4 files

Browse files

Files changed (4) hide show

config_tree.log +139 -0
epoch_019.ckpt +3 -0
metrics.csv +42 -0
tags.log +1 -0

config_tree.log ADDED Viewed

	@@ -0,0 +1,139 @@

+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: 10
+│         step_size: 10
+│         iterate_over_defective_areas: true
+│         save_every_second_good_tile: true
+│
+├── model
+│   └── _target_: src.models.training_module.TrainingLitModule
+│       optimizer:
+│         _target_: torch.optim.Adam
+│         _partial_: true
+│         lr: 0.0019960425587510337
+│         weight_decay: 0.0
+│       scheduler:
+│         _target_: torch.optim.lr_scheduler.ReduceLROnPlateau
+│         _partial_: true
+│         mode: min
+│         factor: 0.1
+│         patience: 10
+│       net:
+│         _target_: src.models.components.cnn_cam_multihead.CNNCAMMultihead
+│         backbone: efficientnet_v2_s
+│         multi_head: false
+│         return_nodes:
+│           features.6.0.block.0: layerout
+│         weights: IMAGENET1K_V1
+│       compile: false
+│
+├── callbacks
+│   └── model_checkpoint:
+│         _target_: lightning.pytorch.callbacks.ModelCheckpoint
+│         dirpath: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-12_14-41-52/0/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-12_14-41-52/0
+│         name: csv/
+│         prefix: ''
+│       wandb:
+│         _target_: lightning.pytorch.loggers.wandb.WandbLogger
+│         save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-12_14-41-52/0
+│         offline: false
+│         id: null
+│         anonymous: null
+│         project: DeepVisionXplain
+│         log_model: false
+│         prefix: ''
+│         entity: team_deepvisionxplain
+│         group: pcb"
+│         tags:
+│         - pcb
+│         - efficientnet_v2_s
+│         - downscaled
+│         job_type: ''
+│       aim:
+│         experiment: efficientnet_v2_s_downscaled_pcb
+│
+├── trainer
+│   └── _target_: lightning.pytorch.trainer.Trainer
+│       default_root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-12_14-41-52/0
+│       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-12_14-41-52/0
+│       work_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain
+│
+├── extras
+│   └── ignore_warnings: false
+│       enforce_tags: true
+│       print_config: true
+│
+├── task_name
+│   └── train
+├── tags
+│   └── ['pcb', 'efficientnet_v2_s', 'downscaled']
+├── train
+│   └── True
+├── test
+│   └── True
+├── ckpt_path
+│   └── None
+├── seed
+│   └── 12345
+└── optimized_metric
+    └── val/acc_best

epoch_019.ckpt ADDED Viewed

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

+version https://git-lfs.github.com/spec/v1
+oid sha256:61bc986634cb091c29efc95b4fa89dad2b03c62c6a9e275a2853258dd809c8e2
+size 66275905

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.9785800576210022,0.9785800576210022,0.09405126422643661
+0,1194,,,0.938464343547821,0.19192448258399963,,,
+1,2389,,,,,0.9810941219329834,0.9810941219329834,0.08577818423509598
+1,2389,,,0.9626811146736145,0.1403200626373291,,,
+2,3584,,,,,0.9801890850067139,0.9810941219329834,0.08877433836460114
+2,3584,,,0.9662377834320068,0.1284174621105194,,,
+3,4779,,,,,0.9799879193305969,0.9810941219329834,0.10236217826604843
+3,4779,,,0.9674146175384521,0.126870259642601,,,
+4,5974,,,,,0.9815969467163086,0.9815969467163086,0.08225452899932861
+4,5974,,,0.9689052700996399,0.12337687611579895,,,
+5,7169,,,,,0.9818986058235168,0.9818986058235168,0.08724121004343033
+5,7169,,,0.9702652096748352,0.1179715171456337,,,
+6,8364,,,,,0.9819992184638977,0.9819992184638977,0.08026734739542007
+6,8364,,,0.9697421193122864,0.11816190183162689,,,
+7,9559,,,,,0.9831053614616394,0.9831053614616394,0.07449725270271301
+7,9559,,,0.9707097411155701,0.11649829894304276,,,
+8,10754,,,,,0.9811946749687195,0.9831053614616394,0.07809150964021683
+8,10754,,,0.9711804986000061,0.11317309737205505,,,
+9,11949,,,,,0.982602596282959,0.9831053614616394,0.07763692736625671
+9,11949,,,0.9712589383125305,0.11222042888402939,,,
+10,13144,,,,,0.9827031493186951,0.9831053614616394,0.07836566120386124
+10,13144,,,0.9718604683876038,0.1116846352815628,,,
+11,14339,,,,,0.982300877571106,0.9831053614616394,0.08225186914205551
+11,14339,,,0.9722527265548706,0.11003268510103226,,,
+12,15534,,,,,0.9811946749687195,0.9831053614616394,0.08842861652374268
+12,15534,,,0.9718865752220154,0.10956701636314392,,,
+13,16729,,,,,0.9836081862449646,0.9836081862449646,0.08435621857643127
+13,16729,,,0.9724096655845642,0.1086176261305809,,,
+14,17924,,,,,0.982602596282959,0.9836081862449646,0.08263920247554779
+14,17924,,,0.9727234840393066,0.10712388902902603,,,
+15,19119,,,,,0.982300877571106,0.9836081862449646,0.07698046416044235
+15,19119,,,0.9728542566299438,0.105719193816185,,,
+16,20314,,,,,0.9834070801734924,0.9836081862449646,0.07577761262655258
+16,20314,,,0.9728803634643555,0.1063806414604187,,,
+17,21509,,,,,0.982602596282959,0.9836081862449646,0.08022763580083847
+17,21509,,,0.9729588627815247,0.10627707093954086,,,
+18,22704,,,,,0.9841110110282898,0.9841110110282898,0.08097345381975174
+18,22704,,,0.9729588627815247,0.1045692190527916,,,
+19,23899,,,,,0.984613835811615,0.984613835811615,0.07302909344434738
+19,23899,,,0.9747371673583984,0.09775423258543015,,,
+20,23900,0.9853643774986267,0.07237692177295685,,,,,

tags.log ADDED Viewed

	@@ -0,0 +1 @@


1	+ ['pcb', 'efficientnet_v2_s', 'downscaled']