DeepVisionXplain
/

efficientnet_v2_s_full_gears

Model card Files Files and versions Community

lus105 commited on Mar 13, 2024

Commit

c9e560f

verified ·

1 Parent(s): 33387ee

Upload 4 files

Browse files

Files changed (4) hide show

config_tree.log +139 -0
epoch_018.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.7: 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_09-06-12/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_09-06-12/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_09-06-12/0
+│         offline: false
+│         id: null
+│         anonymous: null
+│         project: DeepVisionXplain
+│         log_model: false
+│         prefix: ''
+│         entity: team_deepvisionxplain
+│         group: gears
+│         tags:
+│         - gears
+│         - efficientnet_v2_s
+│         - full
+│         job_type: ''
+│       aim:
+│         experiment: efficientnet_v2_s_full_gears
+│
+├── trainer
+│   └── _target_: lightning.pytorch.trainer.Trainer
+│       default_root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-13_09-06-12/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_09-06-12/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', 'full']
+├── train
+│   └── True
+├── test
+│   └── True
+├── ckpt_path
+│   └── None
+├── seed
+│   └── 12345
+└── optimized_metric
+    └── val/acc_best

epoch_018.ckpt ADDED Viewed

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

+version https://git-lfs.github.com/spec/v1
+oid sha256:cbd3ef2f094c718a1e12e8697224d676d1ae0e5413bffea5bd340fbdfefa641c
+size 243474990

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.8840553164482117,0.8840553164482117,0.32100909948349
+0,1432,,,0.8256848454475403,0.4019905626773834,,,
+1,2865,,,,,0.9123502373695374,0.9123502373695374,0.25541287660598755
+1,2865,,,0.8836808800697327,0.2929171621799469,,,
+2,4298,,,,,0.9252534508705139,0.9252534508705139,0.21187911927700043
+2,4298,,,0.9029837846755981,0.25495588779449463,,,
+3,5731,,,,,0.9152073860168457,0.9252534508705139,0.23703855276107788
+3,5731,,,0.9088073372840881,0.23571276664733887,,,
+4,7164,,,,,0.9269124269485474,0.9269124269485474,0.20943181216716766
+4,7164,,,0.9126461148262024,0.22891747951507568,,,
+5,8597,,,,,0.9183410406112671,0.9269124269485474,0.21929749846458435
+5,8597,,,0.9160268902778625,0.22012804448604584,,,
+6,10030,,,,,0.9286636114120483,0.9286636114120483,0.1929755061864853
+6,10030,,,0.9201710224151611,0.20916703343391418,,,
+7,11463,,,,,0.9271889328956604,0.9286636114120483,0.1965666115283966
+7,11463,,,0.9230501055717468,0.20277678966522217,,,
+8,12896,,,,,0.9352995157241821,0.9352995157241821,0.18036526441574097
+8,12896,,,0.9257110357284546,0.19621360301971436,,,
+9,14329,,,,,0.9325345754623413,0.9352995157241821,0.1826656013727188
+9,14329,,,0.9270197153091431,0.19333048164844513,,,
+10,15762,,,,,0.934101402759552,0.9352995157241821,0.18409587442874908
+10,15762,,,0.9273905158042908,0.1888417899608612,,,
+11,17195,,,,,0.9384331703186035,0.9384331703186035,0.173994243144989
+11,17195,,,0.9297897219657898,0.18396992981433868,,,
+12,18628,,,,,0.9360368847846985,0.9384331703186035,0.17713122069835663
+12,18628,,,0.9323634505271912,0.17747792601585388,,,
+13,20061,,,,,0.9342857003211975,0.9384331703186035,0.17308710515499115
+13,20061,,,0.93227618932724,0.17590618133544922,,,
+14,21494,,,,,0.9367741942405701,0.9384331703186035,0.17335502803325653
+14,21494,,,0.9342173933982849,0.1704968959093094,,,
+15,22927,,,,,0.9357603788375854,0.9384331703186035,0.1706673800945282
+15,22927,,,0.9365730285644531,0.16884759068489075,,,
+16,24360,,,,,0.9364055395126343,0.9384331703186035,0.1772608906030655
+16,24360,,,0.9372055530548096,0.16452334821224213,,,
+17,25793,,,,,0.937972366809845,0.9384331703186035,0.1691957414150238
+17,25793,,,0.9389504194259644,0.1628231555223465,,,
+18,27226,,,,,0.9405530095100403,0.9405530095100403,0.1702987104654312
+18,27226,,,0.9389068484306335,0.1588040292263031,,,
+19,28659,,,,,0.9376037120819092,0.9405530095100403,0.16988006234169006
+19,28659,,,0.940455436706543,0.1561402678489685,,,
+20,28660,0.9334116578102112,0.18153607845306396,,,,,

tags.log ADDED Viewed

	@@ -0,0 +1 @@


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