DeepVisionXplain
/

efficientnet_v2_s_downscaled_gears

Model card Files Files and versions Community

lus105 commited on Mar 14, 2024

Commit

9c590ef

verified ·

1 Parent(s): 8e390c9

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: 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-13_23-20-24/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-13_23-20-24/0
+│         name: csv/
+│         prefix: ''
+│       wandb:
+│         _target_: lightning.pytorch.loggers.wandb.WandbLogger
+│         save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-13_23-20-24/0
+│         offline: false
+│         id: null
+│         anonymous: null
+│         project: DeepVisionXplain
+│         log_model: false
+│         prefix: ''
+│         entity: team_deepvisionxplain
+│         group: gears
+│         tags:
+│         - gears
+│         - efficientnet_v2_s
+│         - downscaled
+│         job_type: ''
+│       aim:
+│         experiment: efficientnet_v2_s_downscaled_gears
+│
+├── trainer
+│   └── _target_: lightning.pytorch.trainer.Trainer
+│       default_root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-13_23-20-24/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-13_23-20-24/0
+│       work_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain
+│
+├── extras
+│   └── ignore_warnings: false
+│       enforce_tags: true
+│       print_config: true
+│
+├── task_name
+│   └── train
+├── tags
+│   └── ['gears', '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:849f19c3e353593595050da692756e61a37fa3c08186ba9913ee29223729f6eb
+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,1432,,,,,0.9085714221000671,0.9085714221000671,0.24480490386486053
+0,1432,,,0.8371139168739319,0.37539559602737427,,,
+1,2865,,,,,0.9210138320922852,0.9210138320922852,0.2144012451171875
+1,2865,,,0.8881085515022278,0.2819730341434479,,,
+2,4298,,,,,0.9107834100723267,0.9210138320922852,0.24240484833717346
+2,4298,,,0.9047722816467285,0.24576902389526367,,,
+3,5731,,,,,0.9278340935707092,0.9278340935707092,0.1926228404045105
+3,5731,,,0.911533772945404,0.22922579944133759,,,
+4,7164,,,,,0.9284792542457581,0.9284792542457581,0.2013724148273468
+4,7164,,,0.9162886142730713,0.21806026995182037,,,
+5,8597,,,,,0.9164977073669434,0.9284792542457581,0.23447905480861664
+5,8597,,,0.9194076061248779,0.2108868509531021,,,
+6,10030,,,,,0.9305990934371948,0.9305990934371948,0.1943376362323761
+6,10030,,,0.9221776127815247,0.20505937933921814,,,
+7,11463,,,,,0.9329953789710999,0.9329953789710999,0.19416561722755432
+7,11463,,,0.9232900142669678,0.19831497967243195,,,
+8,12896,,,,,0.9309677481651306,0.9329953789710999,0.19286231696605682
+8,12896,,,0.9265616536140442,0.19185468554496765,,,
+9,14329,,,,,0.9272810816764832,0.9329953789710999,0.21023258566856384
+9,14329,,,0.9285028576850891,0.1868603676557541,,,
+10,15762,,,,,0.9277419447898865,0.9329953789710999,0.2054603397846222
+10,15762,,,0.9310765862464905,0.1829412281513214,,,
+11,17195,,,,,0.9326267242431641,0.9329953789710999,0.1741730272769928
+11,17195,,,0.9313383102416992,0.17788906395435333,,,
+12,18628,,,,,0.9349308609962463,0.9349308609962463,0.17400863766670227
+12,18628,,,0.932952344417572,0.17798283696174622,,,
+13,20061,,,,,0.9214746356010437,0.9349308609962463,0.20284594595432281
+13,20061,,,0.935329794883728,0.16862241923809052,,,
+14,21494,,,,,0.9369585514068604,0.9369585514068604,0.1715237945318222
+14,21494,,,0.9365075826644897,0.16568107903003693,,,
+15,22927,,,,,0.9315207600593567,0.9369585514068604,0.18626686930656433
+15,22927,,,0.9373146295547485,0.16141295433044434,,,
+16,24360,,,,,0.9360368847846985,0.9369585514068604,0.17094112932682037
+16,24360,,,0.9386014938354492,0.15998490154743195,,,
+17,25793,,,,,0.933824896812439,0.9369585514068604,0.18705150485038757
+17,25793,,,0.9403899908065796,0.15627720952033997,,,
+18,27226,,,,,0.9313364028930664,0.9369585514068604,0.18024460971355438
+18,27226,,,0.9410879611968994,0.15260526537895203,,,
+19,28659,,,,,0.9397234916687012,0.9397234916687012,0.16909097135066986
+19,28659,,,0.9422003030776978,0.1503242552280426,,,
+20,28660,0.9356203079223633,0.17211364209651947,,,,,

tags.log ADDED Viewed

	@@ -0,0 +1 @@


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