Add files using upload-large-folder tool
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_depth_head_batch_size=128/lightning_logs/version_0/hparams.yaml +55 -0
- heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_depth_head_batch_size=128/lightning_logs/version_0/results.txt +9 -0
- heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_normals_head_batch_size=128/lightning_logs/version_0/checkpoints/epoch=92-step=2232.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_normals_head_batch_size=128/lightning_logs/version_0/checkpoints/last.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_normals_head_batch_size=128/lightning_logs/version_0/hparams.yaml +55 -0
- heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_segm_head_batch_size=128/lightning_logs/version_0/checkpoints/epoch=41-step=1008.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_segm_head_batch_size=128/lightning_logs/version_0/checkpoints/last.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_pca_depth_head/lightning_logs/version_0/checkpoints/epoch=67-step=1632.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_pca_depth_head/lightning_logs/version_0/checkpoints/last.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_pca_depth_head/lightning_logs/version_0/hparams.yaml +52 -0
- heads/cityscapes/train_cityscapes_224x448_pca_depth_head/lightning_logs/version_0/results.txt +9 -0
- heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_depth_head/lightning_logs/version_0/checkpoints/epoch=59-step=1440.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_depth_head/lightning_logs/version_0/checkpoints/last.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_depth_head/lightning_logs/version_0/hparams.yaml +52 -0
- heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_depth_head/lightning_logs/version_0/results.txt +9 -0
- heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_segm_head/lightning_logs/version_0/checkpoints/epoch=46-step=1128.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_segm_head/lightning_logs/version_0/checkpoints/last.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_segm_head/lightning_logs/version_0/hparams.yaml +52 -0
- heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_segm_head/lightning_logs/version_0/results.txt +2 -0
- heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_surface_normals_head/lightning_logs/version_0/checkpoints/epoch=92-step=2232.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_surface_normals_head/lightning_logs/version_0/checkpoints/last.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_pca_segm_head/lightning_logs/version_0/checkpoints/epoch=51-step=1248.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_pca_segm_head/lightning_logs/version_0/checkpoints/last.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_pca_segm_head/lightning_logs/version_0/hparams.yaml +52 -0
- heads/cityscapes/train_cityscapes_224x448_pca_segm_head/lightning_logs/version_0/results.txt +2 -0
- heads/cityscapes/train_cityscapes_224x448_pca_surface_normals_head/lightning_logs/version_0/checkpoints/epoch=92-step=2232.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_pca_surface_normals_head/lightning_logs/version_0/checkpoints/last.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_pca_surface_normals_head/lightning_logs/version_0/hparams.yaml +52 -0
- heads/cityscapes/train_cityscapes_224x448_pca_surface_normals_head/lightning_logs/version_0/results.txt +8 -0
- heads/cityscapes/train_cityscapes_224x448_raw_depth_head/lightning_logs/version_0/checkpoints/epoch=54-step=1320.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_raw_depth_head/lightning_logs/version_0/checkpoints/last.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_raw_depth_head/lightning_logs/version_0/hparams.yaml +52 -0
- heads/cityscapes/train_cityscapes_224x448_raw_depth_head/lightning_logs/version_0/results.txt +9 -0
- heads/cityscapes/train_cityscapes_224x448_raw_segm_head/lightning_logs/version_0/checkpoints/epoch=59-step=1440.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_raw_segm_head/lightning_logs/version_0/checkpoints/last.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_raw_surface_normals_head/lightning_logs/version_0/checkpoints/epoch=92-step=2232.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_raw_surface_normals_head/lightning_logs/version_0/checkpoints/last.ckpt +3 -0
- heads/cityscapes/train_cityscapes_224x448_raw_surface_normals_head/lightning_logs/version_0/hparams.yaml +52 -0
- heads/cityscapes/train_cityscapes_224x448_raw_surface_normals_head/lightning_logs/version_0/results.txt +8 -0
- heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5/lightning_logs/version_0/checkpoints/epoch=1-step=3640.ckpt +3 -0
- heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5/lightning_logs/version_0/checkpoints/last.ckpt +3 -0
- heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5/lightning_logs/version_0/events.out.tfevents.1760061412.c2-g8-01.4017671.1 +3 -0
- heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5/lightning_logs/version_0/results.txt +9 -0
- heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/checkpoints/epoch=98-step=180180.ckpt +3 -0
- heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/checkpoints/last.ckpt +3 -0
- heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/events.out.tfevents.1760165188.dev2.151251.1 +3 -0
- heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/hparams.yaml +55 -0
- heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5/lightning_logs/version_0/checkpoints/epoch=98-step=180180.ckpt +3 -0
- heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5/lightning_logs/version_0/checkpoints/last.ckpt +3 -0
- heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5/lightning_logs/version_0/events.out.tfevents.1760169054.dev2.132637.1 +3 -0
heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_depth_head_batch_size=128/lightning_logs/version_0/hparams.yaml
ADDED
|
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
args: !!python/object:argparse.Namespace
|
| 2 |
+
accum_iter: 1
|
| 3 |
+
batch_size: 16
|
| 4 |
+
ckpt: null
|
| 5 |
+
data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/cityscapes
|
| 6 |
+
device: !!python/object/apply:torch.device
|
| 7 |
+
- cuda
|
| 8 |
+
- 0
|
| 9 |
+
dlayers:
|
| 10 |
+
- 2
|
| 11 |
+
- 5
|
| 12 |
+
- 8
|
| 13 |
+
- 11
|
| 14 |
+
dpt_out_channels:
|
| 15 |
+
- 128
|
| 16 |
+
- 256
|
| 17 |
+
- 512
|
| 18 |
+
- 512
|
| 19 |
+
dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/cityscapes/heads/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_depth_head_batch_size=128
|
| 20 |
+
effective_batch_size: 128
|
| 21 |
+
eval_ckpt_only: false
|
| 22 |
+
eval_freq: 1
|
| 23 |
+
eval_last: false
|
| 24 |
+
feature_extractor: dino
|
| 25 |
+
gclip: 0.0
|
| 26 |
+
gpu: 0
|
| 27 |
+
hflip_tta: false
|
| 28 |
+
img_size: !!python/tuple
|
| 29 |
+
- 224
|
| 30 |
+
- 448
|
| 31 |
+
lr: 0.0016
|
| 32 |
+
lr_base: 0.0001
|
| 33 |
+
max_epochs: 100
|
| 34 |
+
max_steps: 2325
|
| 35 |
+
modality: depth
|
| 36 |
+
nfeats: 256
|
| 37 |
+
num_classes: 256
|
| 38 |
+
num_gpus: 8
|
| 39 |
+
num_nodes: 1
|
| 40 |
+
num_workers: 8
|
| 41 |
+
num_workers_val: 8
|
| 42 |
+
optimizer: adamw
|
| 43 |
+
pca_ckpt: null
|
| 44 |
+
precision: 16-mixed
|
| 45 |
+
rank: 0
|
| 46 |
+
scheduler: cosine
|
| 47 |
+
seed: 123
|
| 48 |
+
slurm: false
|
| 49 |
+
train_random_photometric_distort: false
|
| 50 |
+
use_bn: true
|
| 51 |
+
use_cls: false
|
| 52 |
+
warmup_p: 0.1
|
| 53 |
+
warmup_steps: 232
|
| 54 |
+
weight_decay: 0.0001
|
| 55 |
+
world_size: 8
|
heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_depth_head_batch_size=128/lightning_logs/version_0/results.txt
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
d1: 0.8475431203842163
|
| 2 |
+
d2: 0.9495058059692383
|
| 3 |
+
d3: 0.9718815088272095
|
| 4 |
+
abs_rel: 0.12315826117992401
|
| 5 |
+
rmse: 7.141202926635742
|
| 6 |
+
rmse_log: 0.24356119334697723
|
| 7 |
+
sq_rel: 1.0250942707061768
|
| 8 |
+
log_10: 0.05426597595214844
|
| 9 |
+
silog: 23.790969848632812
|
heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_normals_head_batch_size=128/lightning_logs/version_0/checkpoints/epoch=92-step=2232.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:983b6933d2e911e4045f7f4febc76f991f1ff49ef9e38f708fe18340ad4ece20
|
| 3 |
+
size 200895759
|
heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_normals_head_batch_size=128/lightning_logs/version_0/checkpoints/last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a97a22c093f26782d684db7b35541efdba42c3a66c36bf1b0a1c1cdd89c8568e
|
| 3 |
+
size 200895759
|
heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_normals_head_batch_size=128/lightning_logs/version_0/hparams.yaml
ADDED
|
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
args: !!python/object:argparse.Namespace
|
| 2 |
+
accum_iter: 1
|
| 3 |
+
batch_size: 16
|
| 4 |
+
ckpt: null
|
| 5 |
+
data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/cityscapes
|
| 6 |
+
device: !!python/object/apply:torch.device
|
| 7 |
+
- cuda
|
| 8 |
+
- 0
|
| 9 |
+
dlayers:
|
| 10 |
+
- 2
|
| 11 |
+
- 5
|
| 12 |
+
- 8
|
| 13 |
+
- 11
|
| 14 |
+
dpt_out_channels:
|
| 15 |
+
- 128
|
| 16 |
+
- 256
|
| 17 |
+
- 512
|
| 18 |
+
- 512
|
| 19 |
+
dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/cityscapes/heads/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_normals_head_batch_size=128
|
| 20 |
+
effective_batch_size: 128
|
| 21 |
+
eval_ckpt_only: false
|
| 22 |
+
eval_freq: 1
|
| 23 |
+
eval_last: false
|
| 24 |
+
feature_extractor: dino
|
| 25 |
+
gclip: 0.0
|
| 26 |
+
gpu: 0
|
| 27 |
+
hflip_tta: false
|
| 28 |
+
img_size: !!python/tuple
|
| 29 |
+
- 224
|
| 30 |
+
- 448
|
| 31 |
+
lr: 0.0016
|
| 32 |
+
lr_base: 0.0001
|
| 33 |
+
max_epochs: 100
|
| 34 |
+
max_steps: 2325
|
| 35 |
+
modality: surface_normals
|
| 36 |
+
nfeats: 256
|
| 37 |
+
num_classes: 3
|
| 38 |
+
num_gpus: 8
|
| 39 |
+
num_nodes: 1
|
| 40 |
+
num_workers: 8
|
| 41 |
+
num_workers_val: 8
|
| 42 |
+
optimizer: adamw
|
| 43 |
+
pca_ckpt: null
|
| 44 |
+
precision: 16-mixed
|
| 45 |
+
rank: 0
|
| 46 |
+
scheduler: poly
|
| 47 |
+
seed: 123
|
| 48 |
+
slurm: false
|
| 49 |
+
train_random_photometric_distort: false
|
| 50 |
+
use_bn: true
|
| 51 |
+
use_cls: false
|
| 52 |
+
warmup_p: 0.1
|
| 53 |
+
warmup_steps: 232
|
| 54 |
+
weight_decay: 0.0001
|
| 55 |
+
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/checkpoints/epoch=41-step=1008.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0e9807eee58f958b8ace9f6b4d81e41294b7ddcd44d98bc01bb25b4c0297e12a
|
| 3 |
+
size 200945103
|
heads/cityscapes/train_cityscapes_224_448_vae_large_l[2,_5,_8,_11]_segm_head_batch_size=128/lightning_logs/version_0/checkpoints/last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1272bf3e2f9e09320aaee94033df9a4753a8c0ffad4431956d1587a0ac523546
|
| 3 |
+
size 200945103
|
heads/cityscapes/train_cityscapes_224x448_pca_depth_head/lightning_logs/version_0/checkpoints/epoch=67-step=1632.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8b3b24bd929eec222f4529f932efd8a88eaa2b5dbbd7ced6776bc34c74f66e97
|
| 3 |
+
size 201674255
|
heads/cityscapes/train_cityscapes_224x448_pca_depth_head/lightning_logs/version_0/checkpoints/last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:49ca3053c14e0b3f386f0215b743dd6c0e4361c4fd1dc73708e61d75ac7c7f45
|
| 3 |
+
size 201674255
|
heads/cityscapes/train_cityscapes_224x448_pca_depth_head/lightning_logs/version_0/hparams.yaml
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
args: !!python/object:argparse.Namespace
|
| 2 |
+
accum_iter: 1
|
| 3 |
+
batch_size: 16
|
| 4 |
+
ckpt: null
|
| 5 |
+
data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/cityscapes
|
| 6 |
+
device: !!python/object/apply:torch.device
|
| 7 |
+
- cuda
|
| 8 |
+
- 0
|
| 9 |
+
dlayers:
|
| 10 |
+
- 2
|
| 11 |
+
- 5
|
| 12 |
+
- 8
|
| 13 |
+
- 11
|
| 14 |
+
dpt_out_channels:
|
| 15 |
+
- 128
|
| 16 |
+
- 256
|
| 17 |
+
- 512
|
| 18 |
+
- 512
|
| 19 |
+
dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/cityscapes/heads/train_cityscapes_224x448_pca_depth_head
|
| 20 |
+
effective_batch_size: 128
|
| 21 |
+
eval_ckpt_only: false
|
| 22 |
+
eval_freq: 1
|
| 23 |
+
eval_last: false
|
| 24 |
+
feature_extractor: dino
|
| 25 |
+
gclip: 0.0
|
| 26 |
+
gpu: 0
|
| 27 |
+
hflip_tta: false
|
| 28 |
+
img_size: !!python/tuple
|
| 29 |
+
- 224
|
| 30 |
+
- 448
|
| 31 |
+
lr: 0.0016
|
| 32 |
+
lr_base: 0.0001
|
| 33 |
+
max_epochs: 100
|
| 34 |
+
max_steps: 2325
|
| 35 |
+
modality: depth
|
| 36 |
+
nfeats: 256
|
| 37 |
+
num_classes: 256
|
| 38 |
+
num_gpus: 8
|
| 39 |
+
num_workers: 8
|
| 40 |
+
num_workers_val: 8
|
| 41 |
+
optimizer: adamw
|
| 42 |
+
pca_ckpt: /users/gabrijel/projects/baselines/official/DINO-Foresight/pcas/cityscapes/dinov2_pca_224_l[2,_5,_8,_11]_1152.pth
|
| 43 |
+
precision: 16-mixed
|
| 44 |
+
rank: 0
|
| 45 |
+
scheduler: cosine
|
| 46 |
+
seed: 123
|
| 47 |
+
use_bn: true
|
| 48 |
+
use_cls: false
|
| 49 |
+
warmup_p: 0.1
|
| 50 |
+
warmup_steps: 232
|
| 51 |
+
weight_decay: 0.0001
|
| 52 |
+
world_size: 8
|
heads/cityscapes/train_cityscapes_224x448_pca_depth_head/lightning_logs/version_0/results.txt
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
d1: 0.867509126663208
|
| 2 |
+
d2: 0.9575021862983704
|
| 3 |
+
d3: 0.9776958227157593
|
| 4 |
+
abs_rel: 0.11600692570209503
|
| 5 |
+
rmse: 6.487044811248779
|
| 6 |
+
rmse_log: 0.21880263090133667
|
| 7 |
+
sq_rel: 0.8949053287506104
|
| 8 |
+
log_10: 0.04848974198102951
|
| 9 |
+
silog: 21.49653434753418
|
heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_depth_head/lightning_logs/version_0/checkpoints/epoch=59-step=1440.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b99eae2b66c50c7ee143d47d241b979328868d0804b527a3ae7e1af4d357b925
|
| 3 |
+
size 201674319
|
heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_depth_head/lightning_logs/version_0/checkpoints/last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b54f063f0ab7e7c9652c1b4da4dcd35e5acf47dae933806c266afa5c0da189bf
|
| 3 |
+
size 201674319
|
heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_depth_head/lightning_logs/version_0/hparams.yaml
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
args: !!python/object:argparse.Namespace
|
| 2 |
+
accum_iter: 1
|
| 3 |
+
batch_size: 16
|
| 4 |
+
ckpt: null
|
| 5 |
+
data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/cityscapes
|
| 6 |
+
device: !!python/object/apply:torch.device
|
| 7 |
+
- cuda
|
| 8 |
+
- 0
|
| 9 |
+
dlayers:
|
| 10 |
+
- 2
|
| 11 |
+
- 5
|
| 12 |
+
- 8
|
| 13 |
+
- 11
|
| 14 |
+
dpt_out_channels:
|
| 15 |
+
- 128
|
| 16 |
+
- 256
|
| 17 |
+
- 512
|
| 18 |
+
- 512
|
| 19 |
+
dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/cityscapes/heads/train_cityscapes_224x448_pca_ncomponents=16_depth_head
|
| 20 |
+
effective_batch_size: 128
|
| 21 |
+
eval_ckpt_only: false
|
| 22 |
+
eval_freq: 1
|
| 23 |
+
eval_last: false
|
| 24 |
+
feature_extractor: dino
|
| 25 |
+
gclip: 0.0
|
| 26 |
+
gpu: 0
|
| 27 |
+
hflip_tta: false
|
| 28 |
+
img_size: !!python/tuple
|
| 29 |
+
- 224
|
| 30 |
+
- 448
|
| 31 |
+
lr: 0.0016
|
| 32 |
+
lr_base: 0.0001
|
| 33 |
+
max_epochs: 100
|
| 34 |
+
max_steps: 2325
|
| 35 |
+
modality: depth
|
| 36 |
+
nfeats: 256
|
| 37 |
+
num_classes: 256
|
| 38 |
+
num_gpus: 8
|
| 39 |
+
num_workers: 8
|
| 40 |
+
num_workers_val: 8
|
| 41 |
+
optimizer: adamw
|
| 42 |
+
pca_ckpt: /work/gabrijel/experiments/dino-foresight/pcas/cityscapes/ncomponents=16/dinov2_pca_224_l[2,_5,_8,_11]_16.pth
|
| 43 |
+
precision: 16-mixed
|
| 44 |
+
rank: 0
|
| 45 |
+
scheduler: cosine
|
| 46 |
+
seed: 123
|
| 47 |
+
use_bn: true
|
| 48 |
+
use_cls: false
|
| 49 |
+
warmup_p: 0.1
|
| 50 |
+
warmup_steps: 232
|
| 51 |
+
weight_decay: 0.0001
|
| 52 |
+
world_size: 8
|
heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_depth_head/lightning_logs/version_0/results.txt
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
d1: 0.815988302230835
|
| 2 |
+
d2: 0.9293420314788818
|
| 3 |
+
d3: 0.9606805443763733
|
| 4 |
+
abs_rel: 0.15756788849830627
|
| 5 |
+
rmse: 8.285951614379883
|
| 6 |
+
rmse_log: 0.29879266023635864
|
| 7 |
+
sq_rel: 1.6426873207092285
|
| 8 |
+
log_10: 0.0647793710231781
|
| 9 |
+
silog: 29.466859817504883
|
heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_segm_head/lightning_logs/version_0/checkpoints/epoch=46-step=1128.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c3f2ce95da37294b53b2bfb84df174ce1cf0a80ccdf723ebdb43d2b04e5916b5
|
| 3 |
+
size 200943311
|
heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_segm_head/lightning_logs/version_0/checkpoints/last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b67a926bb33178f9306c5b351329caa6453f37cce2476cc6fa3f707c21b9b287
|
| 3 |
+
size 200943311
|
heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_segm_head/lightning_logs/version_0/hparams.yaml
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
args: !!python/object:argparse.Namespace
|
| 2 |
+
accum_iter: 1
|
| 3 |
+
batch_size: 16
|
| 4 |
+
ckpt: null
|
| 5 |
+
data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/cityscapes
|
| 6 |
+
device: !!python/object/apply:torch.device
|
| 7 |
+
- cuda
|
| 8 |
+
- 0
|
| 9 |
+
dlayers:
|
| 10 |
+
- 2
|
| 11 |
+
- 5
|
| 12 |
+
- 8
|
| 13 |
+
- 11
|
| 14 |
+
dpt_out_channels:
|
| 15 |
+
- 128
|
| 16 |
+
- 256
|
| 17 |
+
- 512
|
| 18 |
+
- 512
|
| 19 |
+
dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/cityscapes/heads/train_cityscapes_224x448_pca_ncomponents=16_segm_head
|
| 20 |
+
effective_batch_size: 128
|
| 21 |
+
eval_ckpt_only: false
|
| 22 |
+
eval_freq: 1
|
| 23 |
+
eval_last: false
|
| 24 |
+
feature_extractor: dino
|
| 25 |
+
gclip: 0.0
|
| 26 |
+
gpu: 0
|
| 27 |
+
hflip_tta: false
|
| 28 |
+
img_size: !!python/tuple
|
| 29 |
+
- 224
|
| 30 |
+
- 448
|
| 31 |
+
lr: 0.0016
|
| 32 |
+
lr_base: 0.0001
|
| 33 |
+
max_epochs: 100
|
| 34 |
+
max_steps: 2325
|
| 35 |
+
modality: segm
|
| 36 |
+
nfeats: 256
|
| 37 |
+
num_classes: 19
|
| 38 |
+
num_gpus: 8
|
| 39 |
+
num_workers: 8
|
| 40 |
+
num_workers_val: 8
|
| 41 |
+
optimizer: adamw
|
| 42 |
+
pca_ckpt: /work/gabrijel/experiments/dino-foresight/pcas/cityscapes/ncomponents=16/dinov2_pca_224_l[2,_5,_8,_11]_16.pth
|
| 43 |
+
precision: 16-mixed
|
| 44 |
+
rank: 0
|
| 45 |
+
scheduler: poly
|
| 46 |
+
seed: 123
|
| 47 |
+
use_bn: true
|
| 48 |
+
use_cls: false
|
| 49 |
+
warmup_p: 0.1
|
| 50 |
+
warmup_steps: 232
|
| 51 |
+
weight_decay: 0.0001
|
| 52 |
+
world_size: 8
|
heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_segm_head/lightning_logs/version_0/results.txt
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
mIoU: 54.65583801269531
|
| 2 |
+
MO_mIoU: 49.64780807495117
|
heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_surface_normals_head/lightning_logs/version_0/checkpoints/epoch=92-step=2232.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5ff7706b85992d2aa7b2022b096b0085601315cb6a8ee7e56197b6015275c0ee
|
| 3 |
+
size 200894031
|
heads/cityscapes/train_cityscapes_224x448_pca_ncomponents=16_surface_normals_head/lightning_logs/version_0/checkpoints/last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e2e7badb73093fbc19e6b6cb096737bc2b87b29186dd3962ecffec02d61031fa
|
| 3 |
+
size 200894031
|
heads/cityscapes/train_cityscapes_224x448_pca_segm_head/lightning_logs/version_0/checkpoints/epoch=51-step=1248.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:477e701503dde762e8e6706e505aa3f4933daa98f8da0a23921418033d6ca15f
|
| 3 |
+
size 200943247
|
heads/cityscapes/train_cityscapes_224x448_pca_segm_head/lightning_logs/version_0/checkpoints/last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c09184ee00f4e20f4fb2f9e44605eeb575476e901b77fd4568d24ca8ddc47f6a
|
| 3 |
+
size 200943247
|
heads/cityscapes/train_cityscapes_224x448_pca_segm_head/lightning_logs/version_0/hparams.yaml
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
args: !!python/object:argparse.Namespace
|
| 2 |
+
accum_iter: 1
|
| 3 |
+
batch_size: 16
|
| 4 |
+
ckpt: null
|
| 5 |
+
data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/cityscapes
|
| 6 |
+
device: !!python/object/apply:torch.device
|
| 7 |
+
- cuda
|
| 8 |
+
- 0
|
| 9 |
+
dlayers:
|
| 10 |
+
- 2
|
| 11 |
+
- 5
|
| 12 |
+
- 8
|
| 13 |
+
- 11
|
| 14 |
+
dpt_out_channels:
|
| 15 |
+
- 128
|
| 16 |
+
- 256
|
| 17 |
+
- 512
|
| 18 |
+
- 512
|
| 19 |
+
dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/cityscapes/heads/train_cityscapes_224x448_pca_segm_head
|
| 20 |
+
effective_batch_size: 128
|
| 21 |
+
eval_ckpt_only: false
|
| 22 |
+
eval_freq: 1
|
| 23 |
+
eval_last: false
|
| 24 |
+
feature_extractor: dino
|
| 25 |
+
gclip: 0.0
|
| 26 |
+
gpu: 0
|
| 27 |
+
hflip_tta: false
|
| 28 |
+
img_size: !!python/tuple
|
| 29 |
+
- 224
|
| 30 |
+
- 448
|
| 31 |
+
lr: 0.0016
|
| 32 |
+
lr_base: 0.0001
|
| 33 |
+
max_epochs: 100
|
| 34 |
+
max_steps: 2325
|
| 35 |
+
modality: segm
|
| 36 |
+
nfeats: 256
|
| 37 |
+
num_classes: 19
|
| 38 |
+
num_gpus: 8
|
| 39 |
+
num_workers: 8
|
| 40 |
+
num_workers_val: 8
|
| 41 |
+
optimizer: adamw
|
| 42 |
+
pca_ckpt: /users/gabrijel/projects/baselines/official/DINO-Foresight/pcas/cityscapes/dinov2_pca_224_l[2,_5,_8,_11]_1152.pth
|
| 43 |
+
precision: 16-mixed
|
| 44 |
+
rank: 0
|
| 45 |
+
scheduler: poly
|
| 46 |
+
seed: 123
|
| 47 |
+
use_bn: true
|
| 48 |
+
use_cls: false
|
| 49 |
+
warmup_p: 0.1
|
| 50 |
+
warmup_steps: 232
|
| 51 |
+
weight_decay: 0.0001
|
| 52 |
+
world_size: 8
|
heads/cityscapes/train_cityscapes_224x448_pca_segm_head/lightning_logs/version_0/results.txt
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
mIoU: 68.46162414550781
|
| 2 |
+
MO_mIoU: 67.6376953125
|
heads/cityscapes/train_cityscapes_224x448_pca_surface_normals_head/lightning_logs/version_0/checkpoints/epoch=92-step=2232.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:61384f40451c00d5abe11b78f1f77bf539a1f2a5aa4b0c2133938bd716f2d474
|
| 3 |
+
size 200893967
|
heads/cityscapes/train_cityscapes_224x448_pca_surface_normals_head/lightning_logs/version_0/checkpoints/last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f91c408e9ad1d8a870bba2d61fb5289b5a158e43ea40d9ade30296d329fb366f
|
| 3 |
+
size 200893967
|
heads/cityscapes/train_cityscapes_224x448_pca_surface_normals_head/lightning_logs/version_0/hparams.yaml
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
args: !!python/object:argparse.Namespace
|
| 2 |
+
accum_iter: 1
|
| 3 |
+
batch_size: 16
|
| 4 |
+
ckpt: null
|
| 5 |
+
data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/cityscapes
|
| 6 |
+
device: !!python/object/apply:torch.device
|
| 7 |
+
- cuda
|
| 8 |
+
- 0
|
| 9 |
+
dlayers:
|
| 10 |
+
- 2
|
| 11 |
+
- 5
|
| 12 |
+
- 8
|
| 13 |
+
- 11
|
| 14 |
+
dpt_out_channels:
|
| 15 |
+
- 128
|
| 16 |
+
- 256
|
| 17 |
+
- 512
|
| 18 |
+
- 512
|
| 19 |
+
dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/cityscapes/heads/train_cityscapes_224x448_pca_surface_normals_head
|
| 20 |
+
effective_batch_size: 128
|
| 21 |
+
eval_ckpt_only: false
|
| 22 |
+
eval_freq: 1
|
| 23 |
+
eval_last: false
|
| 24 |
+
feature_extractor: dino
|
| 25 |
+
gclip: 0.0
|
| 26 |
+
gpu: 0
|
| 27 |
+
hflip_tta: false
|
| 28 |
+
img_size: !!python/tuple
|
| 29 |
+
- 224
|
| 30 |
+
- 448
|
| 31 |
+
lr: 0.0016
|
| 32 |
+
lr_base: 0.0001
|
| 33 |
+
max_epochs: 100
|
| 34 |
+
max_steps: 2325
|
| 35 |
+
modality: surface_normals
|
| 36 |
+
nfeats: 256
|
| 37 |
+
num_classes: 3
|
| 38 |
+
num_gpus: 8
|
| 39 |
+
num_workers: 8
|
| 40 |
+
num_workers_val: 8
|
| 41 |
+
optimizer: adamw
|
| 42 |
+
pca_ckpt: /users/gabrijel/projects/baselines/official/DINO-Foresight/pcas/cityscapes/dinov2_pca_224_l[2,_5,_8,_11]_1152.pth
|
| 43 |
+
precision: 16-mixed
|
| 44 |
+
rank: 0
|
| 45 |
+
scheduler: poly
|
| 46 |
+
seed: 123
|
| 47 |
+
use_bn: true
|
| 48 |
+
use_cls: false
|
| 49 |
+
warmup_p: 0.1
|
| 50 |
+
warmup_steps: 232
|
| 51 |
+
weight_decay: 0.0001
|
| 52 |
+
world_size: 8
|
heads/cityscapes/train_cityscapes_224x448_pca_surface_normals_head/lightning_logs/version_0/results.txt
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
mean_ae: 2.9138669967651367
|
| 2 |
+
median_ae: 1.7008650302886963
|
| 3 |
+
rmse: 4.548877716064453
|
| 4 |
+
a1: 83.35293579101562
|
| 5 |
+
a2: 92.036865234375
|
| 6 |
+
a3: 97.01936340332031
|
| 7 |
+
a4: 99.61776733398438
|
| 8 |
+
a5: 99.79177856445312
|
heads/cityscapes/train_cityscapes_224x448_raw_depth_head/lightning_logs/version_0/checkpoints/epoch=54-step=1320.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:94cf83072379012e25196036f76fcb88dc7fb4028c11bd19bc392bb0d037d48a
|
| 3 |
+
size 201674127
|
heads/cityscapes/train_cityscapes_224x448_raw_depth_head/lightning_logs/version_0/checkpoints/last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:363628778c6176880c42b44184b13a42c5209526abde6cfe6a1fb60635c468d6
|
| 3 |
+
size 201674127
|
heads/cityscapes/train_cityscapes_224x448_raw_depth_head/lightning_logs/version_0/hparams.yaml
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
args: !!python/object:argparse.Namespace
|
| 2 |
+
accum_iter: 1
|
| 3 |
+
batch_size: 16
|
| 4 |
+
ckpt: null
|
| 5 |
+
data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/cityscapes
|
| 6 |
+
device: !!python/object/apply:torch.device
|
| 7 |
+
- cuda
|
| 8 |
+
- 0
|
| 9 |
+
dlayers:
|
| 10 |
+
- 2
|
| 11 |
+
- 5
|
| 12 |
+
- 8
|
| 13 |
+
- 11
|
| 14 |
+
dpt_out_channels:
|
| 15 |
+
- 128
|
| 16 |
+
- 256
|
| 17 |
+
- 512
|
| 18 |
+
- 512
|
| 19 |
+
dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/cityscapes/heads/train_cityscapes_224x448_raw_depth_head
|
| 20 |
+
effective_batch_size: 128
|
| 21 |
+
eval_ckpt_only: false
|
| 22 |
+
eval_freq: 1
|
| 23 |
+
eval_last: false
|
| 24 |
+
feature_extractor: dino
|
| 25 |
+
gclip: 0.0
|
| 26 |
+
gpu: 0
|
| 27 |
+
hflip_tta: false
|
| 28 |
+
img_size: !!python/tuple
|
| 29 |
+
- 224
|
| 30 |
+
- 448
|
| 31 |
+
lr: 0.0016
|
| 32 |
+
lr_base: 0.0001
|
| 33 |
+
max_epochs: 100
|
| 34 |
+
max_steps: 2325
|
| 35 |
+
modality: depth
|
| 36 |
+
nfeats: 256
|
| 37 |
+
num_classes: 256
|
| 38 |
+
num_gpus: 8
|
| 39 |
+
num_workers: 8
|
| 40 |
+
num_workers_val: 8
|
| 41 |
+
optimizer: adamw
|
| 42 |
+
pca_ckpt: null
|
| 43 |
+
precision: 16-mixed
|
| 44 |
+
rank: 0
|
| 45 |
+
scheduler: cosine
|
| 46 |
+
seed: 123
|
| 47 |
+
use_bn: true
|
| 48 |
+
use_cls: false
|
| 49 |
+
warmup_p: 0.1
|
| 50 |
+
warmup_steps: 232
|
| 51 |
+
weight_decay: 0.0001
|
| 52 |
+
world_size: 8
|
heads/cityscapes/train_cityscapes_224x448_raw_depth_head/lightning_logs/version_0/results.txt
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
d1: 0.8717694282531738
|
| 2 |
+
d2: 0.9581418037414551
|
| 3 |
+
d3: 0.9771357178688049
|
| 4 |
+
abs_rel: 0.11312608420848846
|
| 5 |
+
rmse: 6.49871826171875
|
| 6 |
+
rmse_log: 0.2207271158695221
|
| 7 |
+
sq_rel: 0.8994249701499939
|
| 8 |
+
log_10: 0.047895967960357666
|
| 9 |
+
silog: 21.818635940551758
|
heads/cityscapes/train_cityscapes_224x448_raw_segm_head/lightning_logs/version_0/checkpoints/epoch=59-step=1440.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9875b627808442afdb3a2aed994f8761177678fb99b9813f83d7b8ab0e7a3365
|
| 3 |
+
size 200943119
|
heads/cityscapes/train_cityscapes_224x448_raw_segm_head/lightning_logs/version_0/checkpoints/last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f882719216276271146c0093a46b7f6275e188620b64c50cda48b1f65c4988ee
|
| 3 |
+
size 200943119
|
heads/cityscapes/train_cityscapes_224x448_raw_surface_normals_head/lightning_logs/version_0/checkpoints/epoch=92-step=2232.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b66e1c5912a5245d52e1f0aaa55b9d53901022774d3a02afd02507824bcaafc2
|
| 3 |
+
size 200893839
|
heads/cityscapes/train_cityscapes_224x448_raw_surface_normals_head/lightning_logs/version_0/checkpoints/last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7f21962e574247e82c2ffd67afa981bbc779ca18c8e9059c25d18c3fbf0c2b17
|
| 3 |
+
size 200893839
|
heads/cityscapes/train_cityscapes_224x448_raw_surface_normals_head/lightning_logs/version_0/hparams.yaml
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
args: !!python/object:argparse.Namespace
|
| 2 |
+
accum_iter: 1
|
| 3 |
+
batch_size: 16
|
| 4 |
+
ckpt: null
|
| 5 |
+
data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/cityscapes
|
| 6 |
+
device: !!python/object/apply:torch.device
|
| 7 |
+
- cuda
|
| 8 |
+
- 0
|
| 9 |
+
dlayers:
|
| 10 |
+
- 2
|
| 11 |
+
- 5
|
| 12 |
+
- 8
|
| 13 |
+
- 11
|
| 14 |
+
dpt_out_channels:
|
| 15 |
+
- 128
|
| 16 |
+
- 256
|
| 17 |
+
- 512
|
| 18 |
+
- 512
|
| 19 |
+
dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/cityscapes/heads/train_cityscapes_224x448_raw_surface_normals_head
|
| 20 |
+
effective_batch_size: 128
|
| 21 |
+
eval_ckpt_only: false
|
| 22 |
+
eval_freq: 1
|
| 23 |
+
eval_last: false
|
| 24 |
+
feature_extractor: dino
|
| 25 |
+
gclip: 0.0
|
| 26 |
+
gpu: 0
|
| 27 |
+
hflip_tta: false
|
| 28 |
+
img_size: !!python/tuple
|
| 29 |
+
- 224
|
| 30 |
+
- 448
|
| 31 |
+
lr: 0.0016
|
| 32 |
+
lr_base: 0.0001
|
| 33 |
+
max_epochs: 100
|
| 34 |
+
max_steps: 2325
|
| 35 |
+
modality: surface_normals
|
| 36 |
+
nfeats: 256
|
| 37 |
+
num_classes: 3
|
| 38 |
+
num_gpus: 8
|
| 39 |
+
num_workers: 8
|
| 40 |
+
num_workers_val: 8
|
| 41 |
+
optimizer: adamw
|
| 42 |
+
pca_ckpt: null
|
| 43 |
+
precision: 16-mixed
|
| 44 |
+
rank: 0
|
| 45 |
+
scheduler: poly
|
| 46 |
+
seed: 123
|
| 47 |
+
use_bn: true
|
| 48 |
+
use_cls: false
|
| 49 |
+
warmup_p: 0.1
|
| 50 |
+
warmup_steps: 232
|
| 51 |
+
weight_decay: 0.0001
|
| 52 |
+
world_size: 8
|
heads/cityscapes/train_cityscapes_224x448_raw_surface_normals_head/lightning_logs/version_0/results.txt
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
mean_ae: 2.938774585723877
|
| 2 |
+
median_ae: 1.7030839920043945
|
| 3 |
+
rmse: 4.633583068847656
|
| 4 |
+
a1: 83.20088195800781
|
| 5 |
+
a2: 91.92356872558594
|
| 6 |
+
a3: 96.9259262084961
|
| 7 |
+
a4: 99.578125
|
| 8 |
+
a5: 99.76788330078125
|
heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5/lightning_logs/version_0/checkpoints/epoch=1-step=3640.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:95f1e47d87ae79bcd0ff196a09fd5eebbf30c3078c2732c89f69a7523e11f565
|
| 3 |
+
size 201674255
|
heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5/lightning_logs/version_0/checkpoints/last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f50d9357cf7c511768e1d9ea743c96f64288d68be2276c16d578d60fbf98a82a
|
| 3 |
+
size 201674319
|
heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5/lightning_logs/version_0/events.out.tfevents.1760061412.c2-g8-01.4017671.1
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cad9285c8b9bc6b9c35487723c5f94939f397ea5164d3a99532ac52e57367236
|
| 3 |
+
size 25567
|
heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_depth_head_batch_size=128_lr=1e-5/lightning_logs/version_0/results.txt
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
d1: 0.7751978039741516
|
| 2 |
+
d2: 0.8948086500167847
|
| 3 |
+
d3: 0.9333469271659851
|
| 4 |
+
abs_rel: 0.1531732976436615
|
| 5 |
+
rmse: 3.5306215286254883
|
| 6 |
+
rmse_log: 0.32453784346580505
|
| 7 |
+
sq_rel: 0.7854214906692505
|
| 8 |
+
log_10: 0.0713702142238617
|
| 9 |
+
silog: 31.191608428955078
|
heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/checkpoints/epoch=98-step=180180.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:56276cc1dd1bf2f0f0d672c3709d13b5adc141733ff0e13516ef0432b2546e41
|
| 3 |
+
size 200890895
|
heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/checkpoints/last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:042f87b8a30ea47371ce8174c580d585afc2e7348122ea29249e9d473d8be5a5
|
| 3 |
+
size 200890895
|
heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/events.out.tfevents.1760165188.dev2.151251.1
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a3e26e4bfed8f6fc31f9c5c768b78f4b0ba460aca1039ad77262ba6b0f63a668
|
| 3 |
+
size 7969
|
heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5/lightning_logs/version_0/hparams.yaml
ADDED
|
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
args: !!python/object:argparse.Namespace
|
| 2 |
+
accum_iter: 1
|
| 3 |
+
batch_size: 64
|
| 4 |
+
ckpt: null
|
| 5 |
+
data_path: /scratch/shared/beegfs/gabrijel/datasets/curated/google_kubric
|
| 6 |
+
device: !!python/object/apply:torch.device
|
| 7 |
+
- cuda
|
| 8 |
+
- 0
|
| 9 |
+
dlayers:
|
| 10 |
+
- 2
|
| 11 |
+
- 5
|
| 12 |
+
- 8
|
| 13 |
+
- 11
|
| 14 |
+
dpt_out_channels:
|
| 15 |
+
- 128
|
| 16 |
+
- 256
|
| 17 |
+
- 512
|
| 18 |
+
- 512
|
| 19 |
+
dst_path: /scratch/shared/beegfs/gabrijel/experiments/dino-foresight/logs/kubric/heads/train_kubric_pca_224_l[2,_5,_8,_11]_fgbg_seg_head_batch_size=128_lr=1e-5
|
| 20 |
+
effective_batch_size: 128
|
| 21 |
+
eval_ckpt_only: false
|
| 22 |
+
eval_freq: 1
|
| 23 |
+
eval_last: false
|
| 24 |
+
feature_extractor: dino
|
| 25 |
+
gclip: 0.0
|
| 26 |
+
gpu: 0
|
| 27 |
+
hflip_tta: false
|
| 28 |
+
img_size: !!python/tuple
|
| 29 |
+
- 224
|
| 30 |
+
- 224
|
| 31 |
+
lr: 0.00016
|
| 32 |
+
lr_base: 1.0e-05
|
| 33 |
+
max_epochs: 100
|
| 34 |
+
max_steps: 181950
|
| 35 |
+
modality: segm
|
| 36 |
+
nfeats: 256
|
| 37 |
+
num_classes: 2
|
| 38 |
+
num_gpus: 2
|
| 39 |
+
num_nodes: 1
|
| 40 |
+
num_workers: 8
|
| 41 |
+
num_workers_val: 8
|
| 42 |
+
optimizer: adamw
|
| 43 |
+
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
|
| 44 |
+
precision: 32-true
|
| 45 |
+
rank: 0
|
| 46 |
+
scheduler: poly
|
| 47 |
+
seed: 123
|
| 48 |
+
slurm: false
|
| 49 |
+
train_random_photometric_distort: true
|
| 50 |
+
use_bn: true
|
| 51 |
+
use_cls: false
|
| 52 |
+
warmup_p: 0.1
|
| 53 |
+
warmup_steps: 18195
|
| 54 |
+
weight_decay: 0.0001
|
| 55 |
+
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/checkpoints/epoch=98-step=180180.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:865a378f43328622b4e043c468e44e19019fe0f0bd91e1e8e4b45675e3f18b8c
|
| 3 |
+
size 200894031
|
heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5/lightning_logs/version_0/checkpoints/last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:583321b319f40cebb610be9a53c651f0eb4a90c29f7cc4ea033ab5053354befb
|
| 3 |
+
size 200894031
|
heads/kubric/train_kubric_pca_224_l[2,_5,_8,_11]_normals_head_batch_size=128_lr=1e-5/lightning_logs/version_0/events.out.tfevents.1760169054.dev2.132637.1
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:22ad3e72c1a9d470d65c4c63dbc28535a1eed7097d47ac154dbe5445d9856362
|
| 3 |
+
size 22692
|