bruniss commited on
Commit
fccc4ea
·
verified ·
1 Parent(s): ea35929

Upload 2 files

Browse files
Files changed (2) hide show
  1. config_ps256.json +112 -0
  2. dinov2_ps16_crop256.pt +3 -0
config_ps256.json ADDED
@@ -0,0 +1,112 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "/home/ubuntu/dinovol_2/outputs_ps256",
3
+ "volume_auth": "/home/ubuntu/dinovol_2/creds.json",
4
+ "wandb-entity":"vesuvius-challenge",
5
+ "wandb-project":"dinov2_pretrain",
6
+ "resume": true,
7
+ "resume_from": "/home/ubuntu/dinovol_2/outputs_ps256/checkpoint_step_005000.pt",
8
+ "wandb_resume": "must",
9
+ "wandb_run_id": "vv37qi2n",
10
+ "lr": 2.5e-5,
11
+ "min_lr": 1e-6,
12
+ "weight_decay": 0.04,
13
+ "weight_decay_end": 0.2,
14
+ "patch_embed_lr_mult": 0.2,
15
+ "freeze_last_layer_steps": 1250,
16
+ "momentum_teacher": 0.994,
17
+ "final_momentum_teacher": 1.0,
18
+ "warmup_steps": 35000,
19
+ "warmup_teacher_temp": 0.04,
20
+ "teacher_temp": 0.07,
21
+ "warmup_teacher_temp_steps": 25000,
22
+ "clip_grad": 5.0,
23
+ "max_iterations": 1000000,
24
+ "val_every_n": 500,
25
+ "save_every_n": 2500,
26
+ "batch_size": 3,
27
+ "use_amp": true,
28
+ "centering": "sinkhorn_knopp",
29
+ "num_local_crops": 8,
30
+ "prefetch-factor": 12,
31
+ "dataloader-workers": 12,
32
+ "persistent_workers":true,
33
+ "model": {
34
+ "patch_size": [16, 16, 16]
35
+ },
36
+ "dataset": {
37
+ "source_crop_size": [160, 320, 320],
38
+ "local_crop_scale": [0.02, 0.25],
39
+ "global_crop_size": [
40
+ 128,
41
+ 256,
42
+ 256
43
+ ],
44
+ "local_crop_size": [
45
+ 64,
46
+ 128,
47
+ 128
48
+ ],
49
+ "num_local_crops": 8,
50
+ "volume_auth": "/home/ubuntu/dinovol_2/creds.json",
51
+ "datasets": [
52
+ {
53
+ "volume_path": "s3://vesuvius-challenge-open-data/PHerc0009B/volumes/20250820154339-2.401um-0.3m-77keV-masked.zarr/",
54
+ "volume_scale": 1
55
+ },
56
+ {
57
+ "volume_path": "s3://vesuvius-challenge-open-data/PHerc0500P2/volumes/20250526151718-2.215um-0.4m-111keV-masked.zarr/",
58
+ "volume_scale": 1
59
+ },
60
+ {
61
+ "volume_path": "s3://vesuvius-challenge-open-data/PHerc0814/volumes/20260309142202-2.399um-0.2m-78keV-masked.zarr/",
62
+ "volume_scale": 1
63
+ },
64
+ {
65
+ "volume_path": "s3://vesuvius-challenge-open-data/PHerc1299/volumes/20260309130042-2.399um-0.2m-78keV-masked.zarr/",
66
+ "volume_scale": 1
67
+ },
68
+ {
69
+ "volume_path": "s3://vesuvius-challenge-open-data/PHerc0343P/volumes/20260304131111-2.215um-0.4m-111keV-masked.zarr/",
70
+ "volume_scale": 1
71
+ },
72
+ {
73
+ "volume_path": "s3://vesuvius-challenge-open-data/PHerc0332/volumes/20251211183505-2.399um-0.2m-78keV-masked.zarr/",
74
+ "volume_scale": 1
75
+ },
76
+ {
77
+ "volume_path": "s3://vesuvius-challenge-open-data/PHerc0139/volumes/20260102150214-2.399um-0.2m-78keV-masked.zarr/",
78
+ "volume_scale": 1
79
+ },
80
+ {
81
+ "volume_path": "s3://vesuvius-challenge-open-data/PHercMAN5/volumes/20260311104824-2.399um-0.2m-78keV-masked.zarr/",
82
+ "volume_scale": 1
83
+ }
84
+ ]
85
+ },
86
+ "val_dataset": {
87
+ "source_crop_size": [160, 320, 320],
88
+ "local_crop_scale": [0.02, 0.25],
89
+ "global_crop_size": [
90
+ 128,
91
+ 256,
92
+ 256
93
+ ],
94
+ "local_crop_size": [
95
+ 64,
96
+ 128,
97
+ 128
98
+ ],
99
+ "num_local_crops": 8,
100
+ "volume_auth": "/home/ubuntu/dinovol_2/creds.json",
101
+ "datasets": [
102
+ {
103
+ "volume_path": "s3://vesuvius-challenge-open-data/PHerc0332/volumes/20251211183505-2.399um-0.2m-78keV-masked.zarr/",
104
+ "volume_scale": 1
105
+ },
106
+ {
107
+ "volume_path": "s3://vesuvius-challenge-open-data/PHerc0139/volumes/20260102150214-2.399um-0.2m-78keV-masked.zarr/",
108
+ "volume_scale": 1
109
+ }
110
+ ]
111
+ }
112
+ }
dinov2_ps16_crop256.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d69324e0cbcc0bc1824c2fbdbb2320a557ef4459db0879c3c86ad1f71756982b
3
+ size 5063000261