Hugging Face's logo

DeepVisionXplain
/
mobilenet_v3_large_downscaled_gears

Model card Files Community
lus105 commited on
Commit
fdd57c6
·
verified ·
1 Parent(s): a56bf4a

Upload 4 files

Browse files
Files changed (4) hide show
  1. config_tree.log +139 -0
  2. epoch_017.ckpt +3 -0
  3. metrics.csv +42 -0
  4. tags.log +1 -0
config_tree.log ADDED
@@ -0,0 +1,139 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CONFIG
2
+ ├── data
3
+ │ └── _target_: src.data.dir_datamodule.DirDataModule
4
+ │ data_dir: /media/gpu0/data/lukas/magistras/data/gears/
5
+ │ train_subdir: train/
6
+ │ val_subdir: val/
7
+ │ test_subdir: test/
8
+ │ image_subdir: images/
9
+ │ label_subdir: labels/
10
+ │ batch_size: 64
11
+ │ num_workers: 10
12
+ │ pin_memory: false
13
+ │ preprocessor:
14
+ │ _target_: src.data.components.tile_processor.TilingProcessor
15
+ │ tile_width: 224
16
+ │ tile_height: 224
17
+ │ min_defective_area: 0.01
18
+ │ overlap: 2
19
+ │ step_size: 10
20
+ │ iterate_over_defective_areas: false
21
+ │ save_every_second_good_tile: true
22
+ │
23
+ ├── model
24
+ │ └── _target_: src.models.training_module.TrainingLitModule
25
+ │ optimizer:
26
+ │ _target_: torch.optim.Adam
27
+ │ _partial_: true
28
+ │ lr: 0.0028369961259166574
29
+ │ weight_decay: 0.0
30
+ │ scheduler:
31
+ │ _target_: torch.optim.lr_scheduler.ReduceLROnPlateau
32
+ │ _partial_: true
33
+ │ mode: min
34
+ │ factor: 0.1
35
+ │ patience: 10
36
+ │ net:
37
+ │ _target_: src.models.components.cnn_cam_multihead.CNNCAMMultihead
38
+ │ backbone: mobilenet_v3_large
39
+ │ multi_head: false
40
+ │ return_nodes:
41
+ │ features.13.block.0: layerout
42
+ │ weights: IMAGENET1K_V1
43
+ │ compile: false
44
+ │
45
+ ├── callbacks
46
+ │ └── model_checkpoint:
47
+ │ _target_: lightning.pytorch.callbacks.ModelCheckpoint
48
+ │ dirpath: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-13_01-01-07/2/checkpoints
49
+ │ filename: epoch_{epoch:03d}
50
+ │ monitor: val/acc
51
+ │ verbose: false
52
+ │ save_last: true
53
+ │ save_top_k: 1
54
+ │ mode: max
55
+ │ auto_insert_metric_name: false
56
+ │ save_weights_only: false
57
+ │ every_n_train_steps: null
58
+ │ train_time_interval: null
59
+ │ every_n_epochs: null
60
+ │ save_on_train_epoch_end: null
61
+ │ early_stopping:
62
+ │ _target_: lightning.pytorch.callbacks.EarlyStopping
63
+ │ monitor: val/acc
64
+ │ min_delta: 0.0
65
+ │ patience: 100
66
+ │ verbose: false
67
+ │ mode: max
68
+ │ strict: true
69
+ │ check_finite: true
70
+ │ stopping_threshold: null
71
+ │ divergence_threshold: null
72
+ │ check_on_train_epoch_end: null
73
+ │ model_summary:
74
+ │ _target_: lightning.pytorch.callbacks.RichModelSummary
75
+ │ max_depth: -1
76
+ │ rich_progress_bar:
77
+ │ _target_: lightning.pytorch.callbacks.RichProgressBar
78
+ │
79
+ ├── logger
80
+ │ └── csv:
81
+ │ _target_: lightning.pytorch.loggers.csv_logs.CSVLogger
82
+ │ save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-13_01-01-07/2
83
+ │ name: csv/
84
+ │ prefix: ''
85
+ │ wandb:
86
+ │ _target_: lightning.pytorch.loggers.wandb.WandbLogger
87
+ │ save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-13_01-01-07/2
88
+ │ offline: false
89
+ │ id: null
90
+ │ anonymous: null
91
+ │ project: DeepVisionXplain
92
+ │ log_model: false
93
+ │ prefix: ''
94
+ │ entity: team_deepvisionxplain
95
+ │ group: gears
96
+ │ tags:
97
+ │ - gears
98
+ │ - mobilenet_v3_large
99
+ │ - downscaled
100
+ │ job_type: ''
101
+ │ aim:
102
+ │ experiment: mobilenet_v3_large_full_gears
103
+ │
104
+ ├── trainer
105
+ │ └── _target_: lightning.pytorch.trainer.Trainer
106
+ │ default_root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-13_01-01-07/2
107
+ │ min_epochs: 10
108
+ │ max_epochs: 20
109
+ │ accelerator: gpu
110
+ │ devices: 1
111
+ │ check_val_every_n_epoch: 1
112
+ │ deterministic: false
113
+ │
114
+ ├── paths
115
+ │ └── root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain
116
+ │ data_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/data/
117
+ │ log_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/
118
+ │ output_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-13_01-01-07/2
119
+ │ work_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain
120
+ │
121
+ ├── extras
122
+ │ └── ignore_warnings: false
123
+ │ enforce_tags: true
124
+ │ print_config: true
125
+ │
126
+ ├── task_name
127
+ │ └── train
128
+ ├── tags
129
+ │ └── ['gears', 'mobilenet_v3_large', 'downscaled']
130
+ ├── train
131
+ │ └── True
132
+ ├── test
133
+ │ └── True
134
+ ├── ckpt_path
135
+ │ └── None
136
+ ├── seed
137
+ │ └── 12345
138
+ └── optimized_metric
139
+ └── val/acc_best
epoch_017.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:85c24611a9fa3e687190fa9deb990886f4bb9dee6162220015cf405e66e1c6e7
3
+ size 10843304
metrics.csv ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ epoch,step,test/acc,test/loss,train/acc,train/loss,val/acc,val/acc_best,val/loss
2
+ 0,716,,,,,0.4758525490760803,0.4758525490760803,27.397472381591797
3
+ 0,716,,,0.8276260495185852,0.38907188177108765,,,
4
+ 1,1433,,,,,0.8638709783554077,0.8638709783554077,0.35773134231567383
5
+ 1,1433,,,0.8810198903083801,0.2926620841026306,,,
6
+ 2,2150,,,,,0.9095852375030518,0.9095852375030518,0.2506168484687805
7
+ 2,2150,,,0.8982943892478943,0.25771671533584595,,,
8
+ 3,2867,,,,,0.9094009399414062,0.9095852375030518,0.2564356327056885
9
+ 3,2867,,,0.9047504663467407,0.24231572449207306,,,
10
+ 4,3584,,,,,0.9005529880523682,0.9095852375030518,0.3900590240955353
11
+ 4,3584,,,0.9110757112503052,0.22925922274589539,,,
12
+ 5,4301,,,,,0.9164977073669434,0.9164977073669434,0.2154109925031662
13
+ 5,4301,,,0.914849042892456,0.22252504527568817,,,
14
+ 6,5018,,,,,0.9183410406112671,0.9183410406112671,0.23737873136997223
15
+ 6,5018,,,0.9171610474586487,0.21360361576080322,,,
16
+ 7,5735,,,,,0.9182488322257996,0.9183410406112671,0.2808535099029541
17
+ 7,5735,,,0.9177281260490417,0.21127822995185852,,,
18
+ 8,6452,,,,,0.9179723262786865,0.9183410406112671,0.2035900056362152
19
+ 8,6452,,,0.920759916305542,0.2078848034143448,,,
20
+ 9,7169,,,,,0.9263594746589661,0.9263594746589661,0.20969265699386597
21
+ 9,7169,,,0.9220685958862305,0.20375265181064606,,,
22
+ 10,7886,,,,,0.9300460815429688,0.9300460815429688,0.18200942873954773
23
+ 10,7886,,,0.9234862923622131,0.1997317373752594,,,
24
+ 11,8603,,,,,0.9231336116790771,0.9300460815429688,0.20825771987438202
25
+ 11,8603,,,0.9248386025428772,0.19636325538158417,,,
26
+ 12,9320,,,,,0.9223963022232056,0.9300460815429688,0.21484556794166565
27
+ 12,9320,,,0.9261472821235657,0.1944447010755539,,,
28
+ 13,10037,,,,,0.9086635708808899,0.9300460815429688,0.2582431733608246
29
+ 13,10037,,,0.9268888235092163,0.19235438108444214,,,
30
+ 14,10754,,,,,0.9257143139839172,0.9300460815429688,0.2015533447265625
31
+ 14,10754,,,0.928044855594635,0.18729138374328613,,,
32
+ 15,11471,,,,,0.9209216833114624,0.9300460815429688,0.21760842204093933
33
+ 15,11471,,,0.9301387071609497,0.18366806209087372,,,
34
+ 16,12188,,,,,0.9247004389762878,0.9300460815429688,0.19099006056785583
35
+ 16,12188,,,0.930793046951294,0.18128320574760437,,,
36
+ 17,12905,,,,,0.935944676399231,0.935944676399231,0.17283347249031067
37
+ 17,12905,,,0.9310547709465027,0.17864194512367249,,,
38
+ 18,13622,,,,,0.9266359210014343,0.935944676399231,0.18210071325302124
39
+ 18,13622,,,0.931469202041626,0.17708426713943481,,,
40
+ 19,14339,,,,,0.9254378080368042,0.935944676399231,0.21187426149845123
41
+ 19,14339,,,0.9348499178886414,0.17215751111507416,,,
42
+ 20,14340,0.9318138957023621,0.1696806102991104,,,,,
tags.log ADDED
@@ -0,0 +1 @@
 
 
1
+ ['gears', 'mobilenet_v3_large', 'downscaled']