DeepVisionXplain
/

mobilenet_v3_large_full_gears

Model card Files Files and versions Community

lus105 commited on Mar 12, 2024

Commit

2ee18c8

verified ·

1 Parent(s): 78e6838

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/gears/
+│       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.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: mobilenet_v3_large
+│         multi_head: false
+│         return_nodes:
+│           features.16: 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_19-35-23/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_19-35-23/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_19-35-23/0
+│         offline: false
+│         id: null
+│         anonymous: null
+│         project: DeepVisionXplain
+│         log_model: false
+│         prefix: ''
+│         entity: team_deepvisionxplain
+│         group: gears
+│         tags:
+│         - gears
+│         - mobilenet_v3_large
+│         - full
+│         job_type: ''
+│       aim:
+│         experiment: mobilenet_v3_large_full_gears
+│
+├── trainer
+│   └── _target_: lightning.pytorch.trainer.Trainer
+│       default_root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-12_19-35-23/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_19-35-23/0
+│       work_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain
+│
+├── extras
+│   └── ignore_warnings: false
+│       enforce_tags: true
+│       print_config: true
+│
+├── task_name
+│   └── train
+├── tags
+│   └── ['gears', 'mobilenet_v3_large', 'full']
+├── 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:835ccba881716d46f0b3f68d1e732db2678986eb6047dcec0b50342281293393
+size 36217914

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,1432,,,,,0.689308762550354,0.689308762550354,1.0626293420791626
+0,1432,,,0.8156080842018127,0.41418713331222534,,,
+1,2865,,,,,0.8494930863380432,0.8494930863380432,0.4927027225494385
+1,2865,,,0.8607572913169861,0.32859382033348083,,,
+2,4298,,,,,0.8088479042053223,0.8494930863380432,0.4017987549304962
+2,4298,,,0.8886756300926208,0.281351774930954,,,
+3,5731,,,,,0.9061751365661621,0.9061751365661621,0.26814407110214233
+3,5731,,,0.899603009223938,0.25610554218292236,,,
+4,7164,,,,,0.804423987865448,0.9061751365661621,0.5351445078849792
+4,7164,,,0.9037035703659058,0.24606001377105713,,,
+5,8597,,,,,0.8645161390304565,0.9061751365661621,0.41983407735824585
+5,8597,,,0.9089382290840149,0.23467086255550385,,,
+6,10030,,,,,0.9231336116790771,0.9231336116790771,0.2093438059091568
+6,10030,,,0.9119917750358582,0.23023216426372528,,,
+7,11463,,,,,0.9135484099388123,0.9231336116790771,0.22596049308776855
+7,11463,,,0.9141947031021118,0.2270173877477646,,,
+8,12896,,,,,0.9129953980445862,0.9231336116790771,0.23135623335838318
+8,12896,,,0.9162667989730835,0.22130617499351501,,,
+9,14329,,,,,0.9172350168228149,0.9231336116790771,0.21964974701404572
+9,14329,,,0.9200619459152222,0.2124117612838745,,,
+10,15762,,,,,0.9059907793998718,0.9231336116790771,0.24295571446418762
+10,15762,,,0.9217632412910461,0.20695637166500092,,,
+11,17195,,,,,0.9278340935707092,0.9278340935707092,0.1981792151927948
+11,17195,,,0.9213269948959351,0.204759418964386,,,
+12,18628,,,,,0.9258986115455627,0.9278340935707092,0.2056065946817398
+12,18628,,,0.9246423244476318,0.19857534766197205,,,
+13,20061,,,,,0.9169585108757019,0.9278340935707092,0.22689957916736603
+13,20061,,,0.9249694347381592,0.1970287263393402,,,
+14,21494,,,,,0.9340091943740845,0.9340091943740845,0.17880168557167053
+14,21494,,,0.9257328510284424,0.1948595494031906,,,
+15,22927,,,,,0.9307833909988403,0.9340091943740845,0.18673640489578247
+15,22927,,,0.9276740550994873,0.18949788808822632,,,
+16,24360,,,,,0.9311520457267761,0.9340091943740845,0.20535410940647125
+16,24360,,,0.9275867938995361,0.18899109959602356,,,
+17,25793,,,,,0.9331797361373901,0.9340091943740845,0.1905432939529419
+17,25793,,,0.9298115372657776,0.18550479412078857,,,
+18,27226,,,,,0.9317050576210022,0.9340091943740845,0.17809531092643738
+18,27226,,,0.9287428259849548,0.18440943956375122,,,
+19,28659,,,,,0.9358525276184082,0.9358525276184082,0.21744316816329956
+19,28659,,,0.9301387071609497,0.1819416731595993,,,
+20,28660,0.9332706928253174,0.21472372114658356,,,,,

tags.log ADDED Viewed

	@@ -0,0 +1 @@


1	+ ['gears', 'mobilenet_v3_large', 'full']