Hugging Face's logo

Spaces:
KAIST-Visual-AI-Group
/
salad-demo
Sleeping

App Files Community
2
salad-demo
File size: 875 Bytes 
801501a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
 
_target_: salad.models.phase1.Phase1Model

network:
  _target_: salad.model_components.network.UnCondDiffNetwork
  input_dim: 16
  embedding_dim: 512
  num_heads: 4
  use_timestep_embedder: true
  timestep_embedder_dim: 128
  enc_num_layers: 6
  residual: true
  encoder_type: transformer
  attn_dropout: 0.0

variance_schedule:
  _target_: salad.model_components.variance_schedule.VarianceSchedule
  num_steps: &time_steps 1000
  beta_1: 1e-4
  beta_T: 0.05
  mode: linear

# optimizer
lr: 1e-4
batch_size: 64

# dataset 
dataset_kwargs:
  data_path: spaghetti_chair_latents.hdf5
  repeat: 3
  data_keys: ["g_js_affine"]
  global_normalization: &normalization partial

global_normalization: *normalization # normalize pi, eigenvalues.
num_timesteps: *time_steps
faster: true
validation_step: 10
no_run_validation: false
spaghetti_tag: "chairs_large" # or airplanes, tables