| { |
| "output_dir": "/home/ubuntu/dinovol_2/outputs_try2_basicemb", |
| "volume_auth": "/home/ubuntu/dinovol_2/creds.json", |
| "wandb-entity":"vesuvius-challenge", |
| "wandb-project":"dinov2_pretrain", |
| "resume": true, |
| "resume_from": "/home/ubuntu/dinovol_2/outputs_try2_basicemb/checkpoint_step_227500.pt", |
| "wandb_resume": "must", |
| "wandb_run_id": "1bspskav", |
| "lr": 2.5e-5, |
| "min_lr": 1e-6, |
| "weight_decay": 0.04, |
| "weight_decay_end": 0.2, |
| "patch_embed_lr_mult": 0.2, |
| "freeze_last_layer_steps": 1250, |
| "momentum_teacher": 0.994, |
| "final_momentum_teacher": 1.0, |
| "warmup_steps": 35000, |
| "warmup_teacher_temp": 0.04, |
| "teacher_temp": 0.07, |
| "warmup_teacher_temp_steps": 25000, |
| "clip_grad": 5.0, |
| "max_iterations": 1000000, |
| "val_every_n": 500, |
| "save_every_n": 2500, |
| "batch_size": 3, |
| "use_amp": true, |
| "centering": "sinkhorn_knopp", |
| "num_local_crops": 8, |
| "prefetch-factor": 2, |
| "dataloader-workers": 2, |
| "persistent_workers":true, |
| "model": { |
| "global_crops_size": [ |
| 128, |
| 128, |
| 128 |
| ], |
| "local_crops_size": [ |
| 64, |
| 64, |
| 64 |
| ], |
| "patch_size": [8, 8, 8] |
| }, |
| "dataset": { |
| "source_crop_size": [256, 256, 256], |
| "local_crop_scale": [0.02, 0.25], |
| "global_crop_size": [ |
| 128, |
| 128, |
| 128 |
| ], |
| "local_crop_size": [ |
| 64, |
| 64, |
| 64 |
| ], |
| "num_local_crops": 8, |
| "volume_auth": "/home/ubuntu/dinovol_2/creds.json", |
| "datasets": [ |
| { |
| "volume_path": "s3://vesuvius-challenge-open-data/PHerc0009B/volumes/20250820154339-2.401um-0.3m-77keV-masked.zarr/", |
| "volume_scale": 1 |
| }, |
| { |
| "volume_path": "s3://vesuvius-challenge-open-data/PHerc0500P2/volumes/20250526151718-2.215um-0.4m-111keV-masked.zarr/", |
| "volume_scale": 1 |
| }, |
| { |
| "volume_path": "s3://vesuvius-challenge-open-data/PHerc0814/volumes/20260309142202-2.399um-0.2m-78keV-masked.zarr/", |
| "volume_scale": 1 |
| }, |
| { |
| "volume_path": "s3://vesuvius-challenge-open-data/PHerc1299/volumes/20260309130042-2.399um-0.2m-78keV-masked.zarr/", |
| "volume_scale": 1 |
| }, |
| { |
| "volume_path": "https://volumes.aws.ash2txt.org/esrf/20250717/sam2long-16xdown/SCROLLS_HEL_2.403um_77keV_0.22m_PHerc_0841_B_TA3_0001_masked.zarr/", |
| "volume_scale": 1 |
| }, |
| { |
| "volume_path": "s3://vesuvius-challenge-open-data/PHerc0343P/volumes/20260304131111-2.215um-0.4m-111keV-masked.zarr/", |
| "volume_scale": 1 |
| }, |
| { |
| "volume_path": "https://volumes.aws.ash2txt.org/esrf/20250717/sam2long-16xdown/SCROLLS_HEL_2.403um_77keV_0.22m_PHerc_1203_TA3_0001_masked.zarr/", |
| "volume_scale": 1 |
| }, |
| { |
| "volume_path": "https://volumes.aws.ash2txt.org/esrf/20250717/sam2long-16xdown/SCROLLS_HEL_2.403um_77keV_0.22m_PHerc_0846A_TA3_0001_masked.zarr/", |
| "volume_scale": 1 |
| }, |
| { |
| "volume_path": "https://volumes.aws.ash2txt.org/esrf/20251130/SCROLLS_HEL_2.399um_78keV_0.22m_PHerc_1451_TA2_0001_masked.zarr/", |
| "volume_scale": 1 |
| }, |
| { |
| "volume_path": "s3://vesuvius-challenge-open-data/PHerc0332/volumes/20251211183505-2.399um-0.2m-78keV-masked.zarr/", |
| "volume_scale": 1 |
| }, |
| { |
| "volume_path": "https://volumes.aws.ash2txt.org/esrf/20251130/SCROLLS_HEL_2.399um_78keV_0.22m_PHerc_1667_TA_0001_masked.zarr/", |
| "volume_scale": 1 |
| }, |
| { |
| "volume_path": "https://volumes.aws.ash2txt.org/esrf/20251130/SCROLLS_HEL_2.399um_78keV_0.22m_PHerc_MANBp_bis_TA_0001_masked.zarr/", |
| "volume_scale": 1 |
| } |
| ] |
| }, |
| "val_dataset": { |
| "source_crop_size": [256, 256, 256], |
| "local_crop_scale": [0.02, 0.25], |
| "global_crop_size": [ |
| 128, |
| 128, |
| 128 |
| ], |
| "local_crop_size": [ |
| 64, |
| 64, |
| 64 |
| ], |
| "num_local_crops": 8, |
| "volume_auth": "/home/ubuntu/dinovol_2/creds.json", |
| "datasets": [ |
| { |
| "volume_path": "https://volumes.aws.ash2txt.org/esrf/20250717/sam2long-16xdown/SCROLLS_HEL_2.403um_77keV_0.22m_PHerc_0139_ROI2_TA2_0001_masked.zarr/", |
| "volume_scale": 1 |
| }, |
| { |
| "volume_path": "https://volumes.aws.ash2txt.org/esrf/20251130/SCROLLS_HEL_2.399um_78keV_0.22m_PHerc_1667_TA_0001_masked.zarr/", |
| "volume_scale": 1 |
| } |
| ] |
| } |
| } |
|
|