Spaces:
Running
on
Zero
Running
on
Zero
File size: 866 Bytes
4724018 |
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 |
pc_size: 2048
eval_batch_size: 1
dataloader_num_workers: 4
seed: 0
pred_offset: true
model_type: 'dit_st'
num_inference_steps: 25
model_config:
n_layers: 12
latent_dim: 512
frame_cond: true
point_embed: true
mask_cond: false
pred_offset: true
num_neighbors: -1
floor_cond: true
max_num_forces: 1
force_as_token: true
force_as_latent: false
gravity_emb: true
coeff_cond: false
num_mat: 4
class_token: true
transformer_block: SpatialTemporalTransformerBlock
train_dataset:
category: hf-objaverse-v1
dataset_path: DATASET_FOLDER
dataset_list: DATASET_ITEM_LIST_TEST
has_gravity: true
max_num_forces: ${model_config.max_num_forces}
norm_fac: 5
stage: 'deform'
mode: 'diff'
pc_size: 2048
repeat: 1
seed: 0
n_sample_pro_model: 300
n_frames_interval: 2
n_training_frames: 24
batch_size: 20
overfit: false
|