thanks to NVlabs ❤

lego/config.yaml

@@ -0,0 +1,183 @@
+checkpoint:
+    save_epoch: 9999999999
+    save_iter: 5000
+    save_latest_iter: 9999999999
+    save_period: 9999999999
+    strict_resume: true
+cudnn:
+    benchmark: true
+    deterministic: false
+data:
+    name: dummy
+    num_images: 100
+    num_workers: 4
+    preload: true
+    readjust:
+        center:
+        - 0.0
+        - 0.0
+        - 0.0
+        scale: 1.0
+    root: datasets/lego_skip2/dense
+    train:
+        batch_size: 1
+        image_size:
+        - 799
+        - 799
+        subset: null
+    type: projects.neuralangelo.data
+    use_multi_epoch_loader: true
+    val:
+        batch_size: 1
+        image_size:
+        - 300
+        - 300
+        max_viz_samples: 16
+        subset: 4
+image_save_iter: 9999999999
+inference_args: {}
+local_rank: 0
+logdir: logs/lego_group/lego
+logging_iter: 9999999999999
+max_epoch: 9999999999
+max_iter: 500000
+metrics_epoch: null
+metrics_iter: null
+model:
+    appear_embed:
+        dim: 8
+        enabled: true
+    background:
+        enabled: true
+        encoding:
+            levels: 10
+            type: fourier
+        encoding_view:
+            levels: 3
+            type: spherical
+        mlp:
+            activ: relu
+            activ_density: softplus
+            activ_density_params: {}
+            activ_params: {}
+            hidden_dim: 256
+            hidden_dim_rgb: 128
+            num_layers: 8
+            num_layers_rgb: 2
+            skip:
+            - 4
+            skip_rgb: []
+        view_dep: true
+        white: false
+    object:
+        rgb:
+            encoding_view:
+                levels: 3
+                type: spherical
+            mlp:
+                activ: relu_
+                activ_params: {}
+                hidden_dim: 256
+                num_layers: 4
+                skip: []
+                weight_norm: true
+            mode: idr
+        s_var:
+            anneal_end: 0.1
+            init_val: 3.0
+        sdf:
+            encoding:
+                coarse2fine:
+                    enabled: true
+                    init_active_level: 4
+                    step: 5000
+                hashgrid:
+                    dict_size: 22
+                    dim: 8
+                    max_logres: 11
+                    min_logres: 5
+                    range:
+                    - -2
+                    - 2
+                levels: 16
+                type: hashgrid
+            gradient:
+                mode: numerical
+                taps: 4
+            mlp:
+                activ: softplus
+                activ_params:
+                    beta: 100
+                geometric_init: true
+                hidden_dim: 256
+                inside_out: false
+                num_layers: 1
+                out_bias: 0.5
+                skip: []
+                weight_norm: true
+    render:
+        num_sample_hierarchy: 4
+        num_samples:
+            background: 32
+            coarse: 64
+            fine: 16
+        rand_rays: 512
+        stratified: true
+    type: projects.neuralangelo.model
+nvtx_profile: false
+optim:
+    fused_opt: false
+    params:
+        lr: 0.001
+        weight_decay: 0.01
+    sched:
+        gamma: 10.0
+        iteration_mode: true
+        step_size: 9999999999
+        two_steps:
+        - 300000
+        - 400000
+        type: two_steps_with_warmup
+        warm_up_end: 5000
+    type: AdamW
+pretrained_weight: null
+source_filename: /content/neuralangelo/projects/neuralangelo/configs/custom/lego.yaml
+speed_benchmark: false
+test_data:
+    name: dummy
+    num_workers: 0
+    test:
+        batch_size: 1
+        is_lmdb: false
+        roots: null
+    type: imaginaire.datasets.images
+timeout_period: 9999999
+trainer:
+    amp_config:
+        backoff_factor: 0.5
+        enabled: false
+        growth_factor: 2.0
+        growth_interval: 2000
+        init_scale: 65536.0
+    ddp_config:
+        find_unused_parameters: false
+        static_graph: true
+    depth_vis_scale: 0.5
+    ema_config:
+        beta: 0.9999
+        enabled: false
+        load_ema_checkpoint: false
+        start_iteration: 0
+    grad_accum_iter: 1
+    image_to_tensorboard: false
+    init:
+        gain: null
+        type: none
+    loss_weight:
+        curvature: 0.0005
+        eikonal: 0.1
+        render: 1.0
+    type: projects.neuralangelo.trainer
+validation_iter: 50000
+wandb_image_iter: 1000
+wandb_scalar_iter: 100

lego/epoch_00050_iteration_000005000_checkpoint.pt

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

lego/epoch_00100_iteration_000010000_checkpoint.pt

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

lego/epoch_00150_iteration_000015000_checkpoint.pt

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

lego/latest_checkpoint.txt

@@ -0,0 +1 @@
+epoch_00150_iteration_000015000_checkpoint.pt