DeepVisionXplain
/

mobilenet_v3_large_downscaled_pcb

Model card Files Files and versions Community

lus105 commited on Mar 12, 2024

Commit

161b1ce

verified ·

1 Parent(s): 8ef25b8

Upload 4 files

Browse files

Files changed (4) hide show

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

config_tree.log ADDED Viewed

	@@ -0,0 +1,140 @@

+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: 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.0028369961259166574
+│         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: mobilenet_v3_large
+│         multi_head: false
+│         return_nodes:
+│           features.16: layerout
+│           features.13.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-11_23-45-37/2/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-11_23-45-37/2
+│         name: csv/
+│         prefix: ''
+│       wandb:
+│         _target_: lightning.pytorch.loggers.wandb.WandbLogger
+│         save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-11_23-45-37/2
+│         offline: false
+│         id: null
+│         anonymous: null
+│         project: DeepVisionXplain
+│         log_model: false
+│         prefix: ''
+│         entity: team_deepvisionxplain
+│         group: pcb"
+│         tags:
+│         - pcb
+│         - mobilenet_v3_large
+│         - downscaled
+│         job_type: ''
+│       aim:
+│         experiment: mobilenet_v3_large_pcb_downscaled
+│
+├── trainer
+│   └── _target_: lightning.pytorch.trainer.Trainer
+│       default_root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-11_23-45-37/2
+│       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-11_23-45-37/2
+│       work_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain
+│
+├── extras
+│   └── ignore_warnings: false
+│       enforce_tags: true
+│       print_config: true
+│
+├── task_name
+│   └── train
+├── tags
+│   └── ['pcb', 'mobilenet_v3_large', 'downscaled']
+├── 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:2a907bc9ff762c36530d6a480a6b0b50da479a5289ff07b6aeb3a292963aa169
+size 10843304

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.8836484551429749,0.8836484551429749,0.30515801906585693
+0,597,,,0.8449448347091675,0.35864102840423584,,,
+1,1195,,,,,0.47033387422561646,0.8836484551429749,1.9390511512756348
+1,1195,,,0.8495475649833679,0.3373803198337555,,,
+2,1793,,,,,0.9545454382896423,0.9545454382896423,0.16632792353630066
+2,1793,,,0.9109262824058533,0.24359449744224548,,,
+3,2391,,,,,0.9652051329612732,0.9652051329612732,0.13176055252552032
+3,2391,,,0.9434855580329895,0.1813587248325348,,,
+4,2989,,,,,0.973551869392395,0.973551869392395,0.10721947252750397
+4,2989,,,0.9512003660202026,0.1626519411802292,,,
+5,3587,,,,,0.9729484915733337,0.973551869392395,0.10489927232265472
+5,3587,,,0.9565615653991699,0.148045152425766,,,
+6,4185,,,,,0.9740546941757202,0.9740546941757202,0.10647045075893402
+6,4185,,,0.9580783247947693,0.14708860218524933,,,
+7,4783,,,,,0.9776749610900879,0.9776749610900879,0.09848963469266891
+7,4783,,,0.9587060213088989,0.14709673821926117,,,
+8,5381,,,,,0.9763676524162292,0.9776749610900879,0.10352970659732819
+8,5381,,,0.9603012800216675,0.1413096785545349,,,
+9,5979,,,,,0.975864827632904,0.9776749610900879,0.11164943873882294
+9,5979,,,0.9619227051734924,0.13507114350795746,,,
+10,6577,,,,,0.9699316024780273,0.9776749610900879,0.11432625353336334
+10,6577,,,0.9618180990219116,0.13936513662338257,,,
+11,7175,,,,,0.9739541411399841,0.9776749610900879,0.10410837084054947
+11,7175,,,0.9630995392799377,0.13446000218391418,,,
+12,7773,,,,,0.9739541411399841,0.9776749610900879,0.10635782778263092
+12,7773,,,0.9641979336738586,0.13365283608436584,,,
+13,8371,,,,,0.9770715832710266,0.9776749610900879,0.09549535065889359
+13,8371,,,0.9651132225990295,0.12711001932621002,,,
+14,8969,,,,,0.9651045799255371,0.9776749610900879,0.12173756211996078
+14,8969,,,0.9643809795379639,0.13061146438121796,,,
+15,9567,,,,,0.9772727489471436,0.9776749610900879,0.10027176141738892
+15,9567,,,0.9655055403709412,0.12548476457595825,,,
+16,10165,,,,,0.980088472366333,0.980088472366333,0.08725707232952118
+16,10165,,,0.96566241979599,0.12482082098722458,,,
+17,10763,,,,,0.9809935688972473,0.9809935688972473,0.08831401914358139
+17,10763,,,0.965427041053772,0.12369353324174881,,,
+18,11361,,,,,0.9785800576210022,0.9809935688972473,0.09276366233825684
+18,11361,,,0.9667869806289673,0.12389110028743744,,,
+19,11959,,,,,0.975764274597168,0.9809935688972473,0.120977483689785
+19,11959,,,0.9677022695541382,0.11891140788793564,,,
+20,11960,0.9812325835227966,0.08642280846834183,,,,,

tags.log ADDED Viewed

	@@ -0,0 +1 @@


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