Update config.yaml

config.yaml

@@ -1,5 +1,5 @@
 MODEL:
-  WEIGHTS: /hkfs/work/workspace/scratch/tum_mhj8661-panopticon/dino_logs/abl_a3/v4_16gpu/lr=5e-4_e70b-1_warmup=5/lr=5e-4_warmup=5_hold_epochs=0_min_lr=1e-6_frz=last_layer=1_break_at_epochs=-1/model_final.pth
+  WEIGHTS: path/to/stage1/model_final.pth
   pretrained_weights: []
   WEIGHTS_ONLY: ''
 compute_precision:
@@ -73,24 +73,24 @@ train:
     id: ConcatDataset
     datasets:
     - id: FmowDataset
-      root: /hkfs/work/workspace/scratch/tum_mhj8661-panopticon/datasets/
-      split: /hkfs/work/workspace/scratch/tum_mhj8661-panopticon/datasets/fmow/metadata_v2/fmow_iwm_onid_train_val_savioclean_min42max1024.parquet
+      root: {oc.env:RDIR}/datasets/
+      split: {oc.env:RDIR}/datasets/fmow/metadata_v2/fmow_iwm_onid_train_val_savioclean_min42max1024.parquet
       num_sens: 2
       full_spectra: false
       subset: 800000
     - id: SatlasDataset
-      root: /hkfs/work/workspace/scratch/tum_mhj8661-panopticon/datasets/satlas
+      root: {oc.env:RDIR}/datasets/satlas
       num_sens: 2
       full_spectra: false
-      metadata_path: /hkfs/work/workspace/scratch/tum_mhj8661-panopticon/datasets/satlas/metadata_v2/fmow_iwm_onid_3sensors_all_clean.parquet
+      metadata_path: {oc.env:RDIR}/datasets/satlas/metadata_v2/fmow_iwm_onid_3sensors_all_clean.parquet
     - id: MMEarth
-      base_path: /hkfs/work/workspace/scratch/tum_mhj8661-panopticon/datasets/mmearth/data_1M_v001
+      base_path: {oc.env:RDIR}/datasets/mmearth/data_1M_v001
       split: train
       modalities: MODALITY_MINIMAL_SET1
       full_spectra: false
     - id: SpectralEarth
       split: train
-      root: /hkfs/work/workspace/scratch/tum_mhj8661-panopticon/datasets/
+      root: {oc.env:RDIR}/datasets/
       full_spectra: false
   dino_augm:
     id: ChnSpatialAugmentationV4
@@ -144,14 +144,12 @@ train:
   pin_memory: true
   persistent_workers: true
   use_wandb: true
-  output_dir: /hkfs/work/workspace/scratch/tum_mhj8661-panopticon/dino_logs/abl_a3_s2/v9_16gpu/ds=fmow-800k^satl^mme^spee_lr=5e-4_augm=v4_c13_e=70_model=stg1_16gpu/lr=5e-4_warmup=5_hold_epochs=0_min_lr=1e-6_frz=last_layer=1_break_at_epochs=-1_bsz=75
   saveckp_freq: 5
   seed: 21
   OFFICIAL_EPOCH_LENGTH: 1250
   centering: centering
   log_every_n_steps: 10
   eff_bsz: 75
-  short_exp_name: abl_a3_s2/v9_16gpu/ds=fmow-800k^satl^mme^spee_lr=5e-4_augm=v4_c13_e=70_model=stg1_16gpu
 student:
   arch: vit_base
   embed_layer: ChnFusionPE2