Spaces:
Sleeping
Sleeping
File size: 1,083 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 41 42 43 44 45 46 47 48 49 |
_target_: salad.models.language_phase1.LangPhase1Model
network:
_target_: salad.model_components.network.CondDiffNetwork
input_dim: 16
residual: true
context_dim: 768
context_embedding_dim: 1024
embedding_dim: 512
encoder_use_time: false
encoder_type: pointwise
decoder_type: transformer_encoder
enc_num_layers: 2
dec_num_layers: 6
use_timestep_embedder: true
timestep_embedder_dim: 128
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: 1
data_keys: ["g_js_affine"]
only_easy_context: false
global_normalization: &normalization partial
global_normalization: *normalization
num_timesteps: *time_steps
faster: true
validation_step: 10
no_run_validation: false
spaghetti_tag: "chairs_large" # or airplanes, tables
text_encoder_freeze: false
use_lstm: true
classifier_free_guidance: true
conditioning_dropout_prob: 0.2
|