Add files using upload-large-folder tool

feature_stats/cityscapes/lowres/dino.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c90cbbe51ee0f4f5fd0069cbc4d2c9e8a1212cef37f16614bcac414a95c9ffaa
+size 25076

feature_stats/cityscapes/lowres/raw_convnext_isotropic_base(VAE, beta=0.01).npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3482e32ad26df652c7c3d9d080c048cf116a19308c3fb913ee93589a336fd110
+size 628

feature_stats/cityscapes/lowres/raw_convnext_isotropic_large(VAE, beta=0.01).npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0e8f1706b0a5f5c4e3bc5568818fd153ca75d700ede1bf63b0264ffa03a19232
+size 628

feature_stats/cityscapes/lowres/symmetric_vit_large_registers(VAE, beta=0.01).npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d5d923b21c30e9c922609df3a688d9d7b678fa2f82bc4f5ab251bf020f135a7d
+size 628

feature_stats/ilsvrc/lowres/dino.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:74e58999f69fa0428a20ccbcb5843be55a2dfde266e0087e997f122059441d6f
+size 6644

feature_stats/kubric/movi_a/dino.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b395eb9e709cca2d610c945eee327143e7ffbbb677e76d9c5ffc1acf31ced880
+size 25076

feature_stats/kubric/movi_a/raw_convnext_isotropic_base(VAE, beta=0.01).npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:59498189d4bf95b491e86a0c600739977ec755835e235ee3ecd47adda9517836
+size 628

feature_stats/kubric/movi_a/raw_convnext_isotropic_large(VAE, beta=0.01).npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2bd685bdfaac4af6392bbc597b9db51f46cc9f911217b065f964b0421be8871a
+size 628

feature_stats/vspw/lowres/dino.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:18eff29c2f02b066883108693634b1743c88b5c0a7f410045d07aae3198ef26a
+size 25076

heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_normals_head_batch_size=128/lightning_logs/version_0/events.out.tfevents.1760903611.c2-g8-01.1617721.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ecb3cf920ced923f981c9e5e45aef14f1cad83ac5e223a9218ca1d2692131ff0
+size 315094

heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_normals_head_batch_size=128/lightning_logs/version_0/events.out.tfevents.1760906249.c2-g8-01.1617721.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a92075bcf3215705e8e1e2e8381e9ce61a6ef9a3403b31ff04ca2c03ba7ba0b6
+size 5110

heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_normals_head_batch_size=128/lightning_logs/version_0/results.txt

@@ -0,0 +1,8 @@
+mean_ae: 2.917991876602173
+median_ae: 1.6884536743164062
+rmse: 4.582428932189941
+a1: 83.241455078125
+a2: 91.90704345703125
+a3: 96.94222259521484
+a4: 99.63109588623047
+a5: 99.79322814941406

heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_segm_head_batch_size=128/lightning_logs/version_0/events.out.tfevents.1760897535.c2-g8-01.1415532.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e87f48c4c6e55710fff65b2a0ce3dd13a2aa3b5635b3ee5cb405277156a859dc
+size 136520

heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_segm_head_batch_size=128/lightning_logs/version_0/events.out.tfevents.1760900078.c2-g8-01.1415532.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eb27bcc2fa9fb5aef049c2c3456442a31949e9efb21757d830ac08eb3e91646e
+size 3338

heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_segm_head_batch_size=128/lightning_logs/version_0/hparams.yaml

@@ -0,0 +1,55 @@
+args: !!python/object:argparse.Namespace
+  accum_iter: 1
+  batch_size: 16
+  ckpt: null
+  data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/cityscapes
+  device: !!python/object/apply:torch.device
+  - cuda
+  - 0
+  dlayers:
+  - 2
+  - 5
+  - 8
+  - 11
+  dpt_out_channels:
+  - 128
+  - 256
+  - 512
+  - 512
+  dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/cityscapes/heads/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_segm_head_batch_size=128
+  effective_batch_size: 128
+  eval_ckpt_only: false
+  eval_freq: 1
+  eval_last: false
+  feature_extractor: dino
+  gclip: 0.0
+  gpu: 0
+  hflip_tta: false
+  img_size: !!python/tuple
+  - 224
+  - 448
+  lr: 0.0016
+  lr_base: 0.0001
+  max_epochs: 100
+  max_steps: 2325
+  modality: segm
+  nfeats: 256
+  num_classes: 19
+  num_gpus: 8
+  num_nodes: 1
+  num_workers: 8
+  num_workers_val: 8
+  optimizer: adamw
+  pca_ckpt: null
+  precision: 16-mixed
+  rank: 0
+  scheduler: poly
+  seed: 123
+  slurm: false
+  train_random_photometric_distort: false
+  use_bn: true
+  use_cls: false
+  warmup_p: 0.1
+  warmup_steps: 232
+  weight_decay: 0.0001
+  world_size: 8

heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_segm_head_batch_size=128/lightning_logs/version_0/results.txt

@@ -0,0 +1,2 @@
+mIoU: 65.64169311523438
+MO_mIoU: 64.31041717529297

heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_segm_head/lightning_logs/version_0/events.out.tfevents.1761253552.c2-g8-01.3228201.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ccda849187a5f42acfad8bbfd78da718b9fe07d5da52e28ff621942c1c2977c0
+size 136420

heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_surface_normals_head/lightning_logs/version_0/events.out.tfevents.1761255321.c2-g8-01.3307274.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:877c985ab9d77afdcf05d0575f9c0ab5db33882561fc1653f5c1ae763c1f0bc0
+size 315002

heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_surface_normals_head/lightning_logs/version_0/events.out.tfevents.1761257213.c2-g8-01.3307274.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4de1f102853f0d6e82aace069b48766e1711c31a155e6881dfbadeb8fca1e367
+size 5018

heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_surface_normals_head/lightning_logs/version_0/hparams.yaml

@@ -0,0 +1,52 @@
+args: !!python/object:argparse.Namespace
+  accum_iter: 1
+  batch_size: 16
+  ckpt: null
+  data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/cityscapes
+  device: !!python/object/apply:torch.device
+  - cuda
+  - 0
+  dlayers:
+  - 2
+  - 5
+  - 8
+  - 11
+  dpt_out_channels:
+  - 128
+  - 256
+  - 512
+  - 512
+  dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/cityscapes/heads/train_cityscapes_224x448_pca_ncomponents=16_surface_normals_head
+  effective_batch_size: 128
+  eval_ckpt_only: false
+  eval_freq: 1
+  eval_last: false
+  feature_extractor: dino
+  gclip: 0.0
+  gpu: 0
+  hflip_tta: false
+  img_size: !!python/tuple
+  - 224
+  - 448
+  lr: 0.0016
+  lr_base: 0.0001
+  max_epochs: 100
+  max_steps: 2325
+  modality: surface_normals
+  nfeats: 256
+  num_classes: 3
+  num_gpus: 8
+  num_workers: 8
+  num_workers_val: 8
+  optimizer: adamw
+  pca_ckpt: /work/gabrijel/experiments/dino-foresight/pcas/cityscapes/ncomponents=16/dinov2_pca_224_l[2,_5,_8,_11]_16.pth
+  precision: 16-mixed
+  rank: 0
+  scheduler: poly
+  seed: 123
+  use_bn: true
+  use_cls: false
+  warmup_p: 0.1
+  warmup_steps: 232
+  weight_decay: 0.0001
+  world_size: 8

heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_surface_normals_head/lightning_logs/version_0/results.txt

@@ -0,0 +1,8 @@
+mean_ae: 3.4985973834991455
+median_ae: 2.0157604217529297
+rmse: 5.282929420471191
+a1: 77.00431060791016
+a2: 87.53058624267578
+a3: 94.76048278808594
+a4: 99.57465362548828
+a5: 99.87903594970703

heads/cityscapes/train_cityscapes_224x448_raw_segm_head/lightning_logs/version_0/events.out.tfevents.1760057205.c2-g8-03.1501918.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0965c3c8fa0e09007d3b7b8c766a193e650013c27a923b37a8e9a9e8bcd93183
+size 136297

heads/cityscapes/train_cityscapes_224x448_raw_segm_head/lightning_logs/version_0/events.out.tfevents.1760058634.c2-g8-03.1501918.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:44640782d4f362e4e270cae8ee51f6dc29133ddb6a5d856a9ef723d228ce0dd4
+size 3115

heads/cityscapes/train_cityscapes_224x448_raw_segm_head/lightning_logs/version_0/hparams.yaml

@@ -0,0 +1,52 @@
+args: !!python/object:argparse.Namespace
+  accum_iter: 1
+  batch_size: 16
+  ckpt: null
+  data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/cityscapes
+  device: !!python/object/apply:torch.device
+  - cuda
+  - 0
+  dlayers:
+  - 2
+  - 5
+  - 8
+  - 11
+  dpt_out_channels:
+  - 128
+  - 256
+  - 512
+  - 512
+  dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/cityscapes/heads/train_cityscapes_224x448_raw_segm_head
+  effective_batch_size: 128
+  eval_ckpt_only: false
+  eval_freq: 1
+  eval_last: false
+  feature_extractor: dino
+  gclip: 0.0
+  gpu: 0
+  hflip_tta: false
+  img_size: !!python/tuple
+  - 224
+  - 448
+  lr: 0.0016
+  lr_base: 0.0001
+  max_epochs: 100
+  max_steps: 2325
+  modality: segm
+  nfeats: 256
+  num_classes: 19
+  num_gpus: 8
+  num_workers: 8
+  num_workers_val: 8
+  optimizer: adamw
+  pca_ckpt: null
+  precision: 16-mixed
+  rank: 0
+  scheduler: poly
+  seed: 123
+  use_bn: true
+  use_cls: false
+  warmup_p: 0.1
+  warmup_steps: 232
+  weight_decay: 0.0001
+  world_size: 8

heads/cityscapes/train_cityscapes_224x448_raw_segm_head/lightning_logs/version_0/results.txt

@@ -0,0 +1,2 @@
+mIoU: 68.42170715332031
+MO_mIoU: 66.80571746826172

heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5/lightning_logs/version_0/hparams.yaml

@@ -0,0 +1,55 @@
+args: !!python/object:argparse.Namespace
+  accum_iter: 1
+  batch_size: 32
+  ckpt: null
+  data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/google_kubric
+  device: !!python/object/apply:torch.device
+  - cuda
+  - 0
+  dlayers:
+  - 2
+  - 5
+  - 8
+  - 11
+  dpt_out_channels:
+  - 128
+  - 256
+  - 512
+  - 512
+  dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/kubric/heads/train_kubric_pca_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5
+  effective_batch_size: 128
+  eval_ckpt_only: false
+  eval_freq: 1
+  eval_last: false
+  feature_extractor: dino
+  gclip: 0.0
+  gpu: 0
+  hflip_tta: false
+  img_size: !!python/tuple
+  - 224
+  - 224
+  lr: 0.00016
+  lr_base: 1.0e-05
+  max_epochs: 100
+  max_steps: 181950
+  modality: depth
+  nfeats: 256
+  num_classes: 256
+  num_gpus: 4
+  num_nodes: 1
+  num_workers: 8
+  num_workers_val: 8
+  optimizer: adamw
+  pca_ckpt: /scratch/shared/beegfs/gabrijel/hf/dino-foresight/kubric/pca/subsample_ratio=0.33/dino_pca_224_l[2,_5,_8,_11]_1152.pth
+  precision: 32-true
+  rank: 0
+  scheduler: cosine
+  seed: 123
+  slurm: false
+  train_random_photometric_distort: true
+  use_bn: true
+  use_cls: false
+  warmup_p: 0.1
+  warmup_steps: 18195
+  weight_decay: 0.0001
+  world_size: 4

heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/results.txt

@@ -0,0 +1,2 @@
+mIoU: 99.49028778076172
+MO_mIoU: 99.0764389038086

heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5/lightning_logs/version_0/hparams.yaml

@@ -0,0 +1,55 @@
+args: !!python/object:argparse.Namespace
+  accum_iter: 1
+  batch_size: 64
+  ckpt: null
+  data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/google_kubric
+  device: !!python/object/apply:torch.device
+  - cuda
+  - 0
+  dlayers:
+  - 2
+  - 5
+  - 8
+  - 11
+  dpt_out_channels:
+  - 128
+  - 256
+  - 512
+  - 512
+  dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/kubric/heads/train_kubric_pca_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5
+  effective_batch_size: 128
+  eval_ckpt_only: false
+  eval_freq: 1
+  eval_last: false
+  feature_extractor: dino
+  gclip: 0.0
+  gpu: 0
+  hflip_tta: false
+  img_size: !!python/tuple
+  - 224
+  - 224
+  lr: 0.00016
+  lr_base: 1.0e-05
+  max_epochs: 100
+  max_steps: 181950
+  modality: surface_normals
+  nfeats: 256
+  num_classes: 3
+  num_gpus: 2
+  num_nodes: 1
+  num_workers: 16
+  num_workers_val: 8
+  optimizer: adamw
+  pca_ckpt: /scratch/shared/beegfs/gabrijel/hf/dino-foresight/kubric/pca/subsample_ratio=0.33/dino_pca_224_l[2,_5,_8,_11]_1152.pth
+  precision: 32-true
+  rank: 0
+  scheduler: poly
+  seed: 123
+  slurm: false
+  train_random_photometric_distort: true
+  use_bn: true
+  use_cls: false
+  warmup_p: 0.1
+  warmup_steps: 18195
+  weight_decay: 0.0001
+  world_size: 2

heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5/lightning_logs/version_0/results.txt

@@ -0,0 +1,8 @@
+mean_ae: 0.28686249256134033
+median_ae: 0.0
+rmse: 1.4612072706222534
+a1: 98.61581420898438
+a2: 99.2156753540039
+a3: 99.61569213867188
+a4: 99.93924713134766
+a5: 99.98389434814453

heads/kubric/train_kubric_pca_ncomponents=16_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5/lightning_logs/version_0/events.out.tfevents.1761088937.c2-g8-01.373475.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:151a5f5ac47fd0d3ebf6713aa26c239e6b15aa12cd07cd07ee84a5c4b5e67695
+size 7518447

heads/kubric/train_kubric_pca_ncomponents=16_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/hparams.yaml

@@ -0,0 +1,55 @@
+args: !!python/object:argparse.Namespace
+  accum_iter: 1
+  batch_size: 32
+  ckpt: null
+  data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/google_kubric
+  device: !!python/object/apply:torch.device
+  - cuda
+  - 0
+  dlayers:
+  - 2
+  - 5
+  - 8
+  - 11
+  dpt_out_channels:
+  - 128
+  - 256
+  - 512
+  - 512
+  dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/kubric/heads/train_kubric_pca_ncomponents=16_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5
+  effective_batch_size: 128
+  eval_ckpt_only: false
+  eval_freq: 1
+  eval_last: false
+  feature_extractor: dino
+  gclip: 0.0
+  gpu: 0
+  hflip_tta: false
+  img_size: !!python/tuple
+  - 224
+  - 224
+  lr: 0.00016
+  lr_base: 1.0e-05
+  max_epochs: 100
+  max_steps: 181950
+  modality: segm
+  nfeats: 256
+  num_classes: 2
+  num_gpus: 4
+  num_nodes: 1
+  num_workers: 8
+  num_workers_val: 8
+  optimizer: adamw
+  pca_ckpt: /work/gabrijel/experiments/dino-foresight/pcas/kubric/ncomponents=16/dino_pca_224_l[2,_5,_8,_11]_16.pth
+  precision: 16-mixed
+  rank: 0
+  scheduler: poly
+  seed: 123
+  slurm: false
+  train_random_photometric_distort: false
+  use_bn: true
+  use_cls: false
+  warmup_p: 0.1
+  warmup_steps: 18195
+  weight_decay: 0.0001
+  world_size: 4

heads/kubric/train_kubric_pca_ncomponents=16_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/results.txt

@@ -0,0 +1,2 @@
+mIoU: 97.4259033203125
+MO_mIoU: 95.34408569335938

heads/kubric/train_kubric_pca_ncomponents=16_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5/lightning_logs/version_0/hparams.yaml

@@ -0,0 +1,55 @@
+args: !!python/object:argparse.Namespace
+  accum_iter: 1
+  batch_size: 32
+  ckpt: null
+  data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/google_kubric
+  device: !!python/object/apply:torch.device
+  - cuda
+  - 0
+  dlayers:
+  - 2
+  - 5
+  - 8
+  - 11
+  dpt_out_channels:
+  - 128
+  - 256
+  - 512
+  - 512
+  dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/kubric/heads/train_kubric_pca_ncomponents=16_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5
+  effective_batch_size: 128
+  eval_ckpt_only: false
+  eval_freq: 1
+  eval_last: false
+  feature_extractor: dino
+  gclip: 0.0
+  gpu: 0
+  hflip_tta: false
+  img_size: !!python/tuple
+  - 224
+  - 224
+  lr: 0.00016
+  lr_base: 1.0e-05
+  max_epochs: 100
+  max_steps: 181950
+  modality: surface_normals
+  nfeats: 256
+  num_classes: 3
+  num_gpus: 4
+  num_nodes: 1
+  num_workers: 8
+  num_workers_val: 8
+  optimizer: adamw
+  pca_ckpt: /work/gabrijel/experiments/dino-foresight/pcas/kubric/ncomponents=16/dino_pca_224_l[2,_5,_8,_11]_16.pth
+  precision: 16-mixed
+  rank: 0
+  scheduler: poly
+  seed: 123
+  slurm: false
+  train_random_photometric_distort: false
+  use_bn: true
+  use_cls: false
+  warmup_p: 0.1
+  warmup_steps: 18195
+  weight_decay: 0.0001
+  world_size: 4

heads/kubric/train_kubric_pca_ncomponents=16_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5/lightning_logs/version_0/results.txt

@@ -0,0 +1,8 @@
+mean_ae: 0.4457050561904907
+median_ae: 0.0
+rmse: 2.2965760231018066
+a1: 97.7518081665039
+a2: 98.4872817993164
+a3: 99.06644439697266
+a4: 99.74061584472656
+a5: 99.8936996459961

heads/kubric/train_kubric_raw_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5/lightning_logs/version_0/hparams.yaml

@@ -0,0 +1,55 @@
+args: !!python/object:argparse.Namespace
+  accum_iter: 1
+  batch_size: 32
+  ckpt: null
+  data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/google_kubric
+  device: !!python/object/apply:torch.device
+  - cuda
+  - 0
+  dlayers:
+  - 2
+  - 5
+  - 8
+  - 11
+  dpt_out_channels:
+  - 128
+  - 256
+  - 512
+  - 512
+  dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/kubric/heads/train_kubric_raw_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5
+  effective_batch_size: 128
+  eval_ckpt_only: false
+  eval_freq: 1
+  eval_last: false
+  feature_extractor: dino
+  gclip: 0.0
+  gpu: 0
+  hflip_tta: false
+  img_size: !!python/tuple
+  - 224
+  - 224
+  lr: 0.00016
+  lr_base: 1.0e-05
+  max_epochs: 100
+  max_steps: 181950
+  modality: depth
+  nfeats: 256
+  num_classes: 256
+  num_gpus: 4
+  num_nodes: 1
+  num_workers: 8
+  num_workers_val: 8
+  optimizer: adamw
+  pca_ckpt: null
+  precision: 32-true
+  rank: 0
+  scheduler: cosine
+  seed: 123
+  slurm: false
+  train_random_photometric_distort: true
+  use_bn: true
+  use_cls: false
+  warmup_p: 0.1
+  warmup_steps: 18195
+  weight_decay: 0.0001
+  world_size: 4

heads/kubric/train_kubric_raw_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5/lightning_logs/version_0/results.txt

@@ -0,0 +1,9 @@
+d1: 0.7762729525566101
+d2: 0.8912768959999084
+d3: 0.9289162755012512
+abs_rel: 0.15639297664165497
+rmse: 3.480448007583618
+rmse_log: 0.3384455740451813
+sq_rel: 0.781800389289856
+log_10: 0.07297655940055847
+silog: 32.50920104980469

heads/kubric/train_kubric_raw_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/events.out.tfevents.1760186478.c2-g8-05.1270284.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:831e43e2ce79921e4b18eb1adadac25d82091f888240f6de4a604453f53b2827
+size 7854

heads/kubric/train_kubric_raw_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/hparams.yaml

@@ -0,0 +1,55 @@
+args: !!python/object:argparse.Namespace
+  accum_iter: 1
+  batch_size: 64
+  ckpt: null
+  data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/google_kubric
+  device: !!python/object/apply:torch.device
+  - cuda
+  - 0
+  dlayers:
+  - 2
+  - 5
+  - 8
+  - 11
+  dpt_out_channels:
+  - 128
+  - 256
+  - 512
+  - 512
+  dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/kubric/heads/train_kubric_raw_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5
+  effective_batch_size: 128
+  eval_ckpt_only: false
+  eval_freq: 1
+  eval_last: false
+  feature_extractor: dino
+  gclip: 0.0
+  gpu: 0
+  hflip_tta: false
+  img_size: !!python/tuple
+  - 224
+  - 224
+  lr: 0.00016
+  lr_base: 1.0e-05
+  max_epochs: 100
+  max_steps: 181950
+  modality: segm
+  nfeats: 256
+  num_classes: 2
+  num_gpus: 2
+  num_nodes: 1
+  num_workers: 8
+  num_workers_val: 8
+  optimizer: adamw
+  pca_ckpt: null
+  precision: 32-true
+  rank: 0
+  scheduler: poly
+  seed: 123
+  slurm: false
+  train_random_photometric_distort: true
+  use_bn: true
+  use_cls: false
+  warmup_p: 0.1
+  warmup_steps: 18195
+  weight_decay: 0.0001
+  world_size: 2

heads/kubric/train_kubric_raw_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/results.txt

@@ -0,0 +1,2 @@
+mIoU: 99.53663635253906
+MO_mIoU: 99.160400390625

heads/kubric/train_kubric_vae_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5/lightning_logs/version_0/events.out.tfevents.1760742898.c2-g8-01.1769812.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:64b45a91c1d6b02a271fbdedb4c1c236ab3d696a8415b11656d69efbfea04da1
+size 25453

heads/kubric/train_kubric_vae_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5/lightning_logs/version_0/results.txt

@@ -0,0 +1,9 @@
+d1: 0.7801956534385681
+d2: 0.8996610641479492
+d3: 0.9364328384399414
+abs_rel: 0.15711060166358948
+rmse: 3.606865167617798
+rmse_log: 0.31699931621551514
+sq_rel: 0.8544824719429016
+log_10: 0.06970246881246567
+silog: 30.69348907470703

heads/kubric/train_kubric_vae_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/checkpoints/epoch=43-step=80080.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f0a5233dd37c128f0a62dabe48d64e8d2b79bf67b38738349859830e5e4ac368
+size 200891855

heads/kubric/train_kubric_vae_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/events.out.tfevents.1760732291.dev2.2850877.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8e848a2e68e5562849c7faf1bf4a7eb07133bdb314a5214c912cf1c3b083dc6e
+size 7855

heads/kubric/train_kubric_vae_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/hparams.yaml

@@ -0,0 +1,55 @@
+args: !!python/object:argparse.Namespace
+  accum_iter: 1
+  batch_size: 32
+  ckpt: null
+  data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/google_kubric
+  device: !!python/object/apply:torch.device
+  - cuda
+  - 0
+  dlayers:
+  - 2
+  - 5
+  - 8
+  - 11
+  dpt_out_channels:
+  - 128
+  - 256
+  - 512
+  - 512
+  dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/kubric/heads/train_kubric_vae_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5
+  effective_batch_size: 128
+  eval_ckpt_only: false
+  eval_freq: 1
+  eval_last: false
+  feature_extractor: dino
+  gclip: 0.0
+  gpu: 0
+  hflip_tta: false
+  img_size: !!python/tuple
+  - 224
+  - 224
+  lr: 0.00016
+  lr_base: 1.0e-05
+  max_epochs: 100
+  max_steps: 181950
+  modality: segm
+  nfeats: 256
+  num_classes: 2
+  num_gpus: 4
+  num_nodes: 1
+  num_workers: 8
+  num_workers_val: 8
+  optimizer: adamw
+  pca_ckpt: null
+  precision: 16-mixed
+  rank: 0
+  scheduler: poly
+  seed: 123
+  slurm: false
+  train_random_photometric_distort: false
+  use_bn: true
+  use_cls: false
+  warmup_p: 0.1
+  warmup_steps: 18195
+  weight_decay: 0.0001
+  world_size: 4

heads/kubric/train_kubric_vae_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/results.txt

@@ -0,0 +1,2 @@
+mIoU: 99.16780853271484
+MO_mIoU: 98.49254608154297

heads/kubric/train_kubric_vae_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5/lightning_logs/version_0/checkpoints/epoch=99-step=182000.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:42d99fc1bf37578a9df34f5ab34396024e44c4268d541408e7eb4b0a07dcb389
+size 200894927

heads/kubric/train_kubric_vae_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5/lightning_logs/version_0/checkpoints/last.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:42d99fc1bf37578a9df34f5ab34396024e44c4268d541408e7eb4b0a07dcb389
+size 200894927

heads/kubric/train_kubric_vae_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5/lightning_logs/version_0/events.out.tfevents.1760720872.c2-g4-13.1359983.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a82a961848a481a9449710c05f7b6a2acfc2a908417fefe37f66debd66b688bb
+size 22578

heads/kubric/train_kubric_vae_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5/lightning_logs/version_0/hparams.yaml

@@ -0,0 +1,55 @@
+args: !!python/object:argparse.Namespace
+  accum_iter: 1
+  batch_size: 32
+  ckpt: null
+  data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/google_kubric
+  device: !!python/object/apply:torch.device
+  - cuda
+  - 0
+  dlayers:
+  - 2
+  - 5
+  - 8
+  - 11
+  dpt_out_channels:
+  - 128
+  - 256
+  - 512
+  - 512
+  dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/kubric/heads/train_kubric_vae_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5
+  effective_batch_size: 128
+  eval_ckpt_only: false
+  eval_freq: 1
+  eval_last: false
+  feature_extractor: dino
+  gclip: 0.0
+  gpu: 0
+  hflip_tta: false
+  img_size: !!python/tuple
+  - 224
+  - 224
+  lr: 0.00016
+  lr_base: 1.0e-05
+  max_epochs: 100
+  max_steps: 181950
+  modality: surface_normals
+  nfeats: 256
+  num_classes: 3
+  num_gpus: 4
+  num_nodes: 1
+  num_workers: 8
+  num_workers_val: 8
+  optimizer: adamw
+  pca_ckpt: null
+  precision: 16-mixed
+  rank: 0
+  scheduler: poly
+  seed: 123
+  slurm: false
+  train_random_photometric_distort: false
+  use_bn: true
+  use_cls: false
+  warmup_p: 0.1
+  warmup_steps: 18195
+  weight_decay: 0.0001
+  world_size: 4

heads/kubric/train_kubric_vae_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5/lightning_logs/version_0/results.txt

@@ -0,0 +1,8 @@
+mean_ae: 0.302557110786438
+median_ae: 0.0
+rmse: 1.575514793395996
+a1: 98.54328155517578
+a2: 99.14720916748047
+a3: 99.55836486816406
+a4: 99.91535186767578
+a5: 99.97205352783203