DeepVisionXplain
/

vit_tiny_patch16_224.augreg_in21k_ft_in1k_gears

Model card Files Files and versions Community

lus105 commited on Mar 16, 2024

Commit

21e6d05

verified ·

1 Parent(s): 3623214

Upload 4 files

Browse files

Files changed (4) hide show

config_tree.log +138 -0
epoch_010.ckpt +3 -0
metrics.csv +42 -0
tags.log +1 -0

config_tree.log ADDED Viewed

	@@ -0,0 +1,138 @@

+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.AdamW
+│         _partial_: true
+│         lr: 7.955439594511367e-05
+│         weight_decay: 0.0
+│       scheduler: null
+│       net:
+│         _target_: src.models.components.vit_rollout_multihead.VitRolloutMultihead
+│         model_name: vit_tiny_patch16_224.augreg_in21k_ft_in1k
+│         pretrained: true
+│         weight_path: null
+│         output_size: 1
+│         return_nodes: attn_drop
+│         head_name: head
+│         img_size: 224
+│         discard_ratio: 0.2
+│         head_fusion: mean
+│         visualize: false
+│       compile: false
+│
+├── callbacks
+│   └── model_checkpoint:
+│         _target_: lightning.pytorch.callbacks.ModelCheckpoint
+│         dirpath: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-16_11-55-25/6/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-16_11-55-25/6
+│         name: csv/
+│         prefix: ''
+│       wandb:
+│         _target_: lightning.pytorch.loggers.wandb.WandbLogger
+│         save_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-16_11-55-25/6
+│         offline: false
+│         id: null
+│         anonymous: null
+│         project: DeepVisionXplain
+│         log_model: false
+│         prefix: ''
+│         entity: team_deepvisionxplain
+│         group: gears
+│         tags:
+│         - gears
+│         - vit_tiny_patch16_224.augreg_in21k_ft_in1k
+│         job_type: ''
+│       aim:
+│         experiment: vit_tiny_patch16_224.augreg_in21k_ft_in1k_gears
+│
+├── trainer
+│   └── _target_: lightning.pytorch.trainer.Trainer
+│       default_root_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain/logs/train/multiruns/2024-03-16_11-55-25/6
+│       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-16_11-55-25/6
+│       work_dir: /media/gpu0/data/lukas/magistras/DeepVisionXplain
+│
+├── extras
+│   └── ignore_warnings: false
+│       enforce_tags: true
+│       print_config: true
+│
+├── task_name
+│   └── train
+├── tags
+│   └── ['gears', 'vit_tiny_patch16_224.augreg_in21k_ft_in1k']
+├── train
+│   └── True
+├── test
+│   └── True
+├── ckpt_path
+│   └── None
+├── seed
+│   └── 12345
+└── optimized_metric
+    └── val/acc_best

epoch_010.ckpt ADDED Viewed

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

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

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.9325345754623413,0.9325345754623413,0.19918356835842133
+0,1432,,,0.9039871096611023,0.24156254529953003,,,
+1,2865,,,,,0.9353917241096497,0.9353917241096497,0.1776854693889618
+1,2865,,,0.9277395009994507,0.18794547021389008,,,
+2,4298,,,,,0.9330875873565674,0.9353917241096497,0.17705915868282318
+2,4298,,,0.9346100091934204,0.17284728586673737,,,
+3,5731,,,,,0.9301382303237915,0.9353917241096497,0.19981296360492706
+3,5731,,,0.938077986240387,0.16172367334365845,,,
+4,7164,,,,,0.9270967841148376,0.9353917241096497,0.1936749368906021
+4,7164,,,0.9415241479873657,0.1525978296995163,,,
+5,8597,,,,,0.9388018250465393,0.9388018250465393,0.17147229611873627
+5,8597,,,0.9452974796295166,0.14374929666519165,,,
+6,10030,,,,,0.92580646276474,0.9388018250465393,0.19466955959796906
+6,10030,,,0.9472604990005493,0.13786929845809937,,,
+7,11463,,,,,0.9304147362709045,0.9388018250465393,0.18817198276519775
+7,11463,,,0.9495506882667542,0.13097579777240753,,,
+8,12896,,,,,0.9314285516738892,0.9388018250465393,0.1952507048845291
+8,12896,,,0.9522116780281067,0.1263027787208557,,,
+9,14329,,,,,0.9364055395126343,0.9388018250465393,0.1802290827035904
+9,14329,,,0.95491623878479,0.11994071304798126,,,
+10,15762,,,,,0.9420276284217834,0.9420276284217834,0.16476842761039734
+10,15762,,,0.9566829800605774,0.11337895691394806,,,
+11,17195,,,,,0.9366820454597473,0.9420276284217834,0.16678594052791595
+11,17195,,,0.9587986469268799,0.1090051457285881,,,
+12,18628,,,,,0.9385253190994263,0.9420276284217834,0.17359019815921783
+12,18628,,,0.9609797596931458,0.10527420043945312,,,
+13,20061,,,,,0.9388940334320068,0.9420276284217834,0.18199294805526733
+13,20061,,,0.9614160060882568,0.099303238093853,,,
+14,21494,,,,,0.936497688293457,0.9420276284217834,0.17797796428203583
+14,21494,,,0.9645349979400635,0.09407127648591995,,,
+15,22927,,,,,0.9363133907318115,0.9420276284217834,0.17959438264369965
+15,22927,,,0.9653856158256531,0.09136372804641724,,,
+16,24360,,,,,0.9353917241096497,0.9420276284217834,0.19542205333709717
+16,24360,,,0.9677194356918335,0.0856977328658104,,,
+17,25793,,,,,0.9388018250465393,0.9420276284217834,0.19473664462566376
+17,25793,,,0.9669778347015381,0.08378621190786362,,,
+18,27226,,,,,0.9348387122154236,0.9420276284217834,0.2037130445241928
+18,27226,,,0.9699223637580872,0.07796860486268997,,,
+19,28659,,,,,0.9395391941070557,0.9420276284217834,0.20277413725852966
+19,28659,,,0.9717981219291687,0.07642525434494019,,,
+20,28660,0.9373590350151062,0.17653676867485046,,,,,

tags.log ADDED Viewed

	@@ -0,0 +1 @@


1	+ ['gears', 'vit_tiny_patch16_224.augreg_in21k_ft_in1k']