Upload multi-category shapenet model

config_nmr.yaml

@@ -0,0 +1,66 @@
+wandb:
+  project: gs_pred
+cam_embd:
+  embedding: null
+  encode_embedding: null
+  dimension: 0
+  method: null
+general:
+  device: 0
+  random_seed: 0
+data:
+  training_resolution: 64
+  fov: 29.999999290482176
+  subset: -1
+  input_images: 1
+  znear: 1.2
+  zfar: 4.0
+  category: nmr
+  white_background: true
+  origin_distances: false
+opt:
+  iterations: 1390001
+  base_lr: 5.0e-05
+  batch_size: 32
+  betas:
+  - 0.9
+  - 0.999
+  loss: l2
+  imgs_per_obj: 4
+  ema:
+    use: true
+    update_every: 10
+    update_after_step: 100
+    beta: 0.9999
+  lambda_lpips: 0.01
+  start_lpips_after: 1240001
+  step_lr_at: 1240001
+model:
+  max_sh_degree: 1
+  inverted_x: false
+  inverted_y: true
+  name: SingleUNet
+  opacity_scale: 1.0
+  opacity_bias: -2.0
+  scale_scale: 0.01
+  scale_bias: 0.02
+  xyz_scale: 0.1
+  xyz_bias: 0.0
+  depth_scale: 1.0
+  depth_bias: 0.0
+  features_dc_scale: 5.0
+  features_sh_scale: 0.0
+  network_without_offset: false
+  network_with_offset: true
+  attention_resolutions:
+  - 8
+  num_blocks: 4
+  cross_view_attention: true
+  base_dim: 128
+  isotropic: false
+logging:
+  ckpt_iterations: 1000
+  val_log: 10000
+  loss_log: 10
+  loop_log: 10000
+  render_log: 10000

model_nmr.pth

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