File size: 2,074 Bytes
ae5426a |
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 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 |
exp_name: optimization_cube
seed: &seed 2010445
out_dir: ./logs/
tensorboard_log_dir: ./logs/optimization_cube
save_every_iter: &log_freq 10
total_steps: 1024 # Check diffusion_cfg to see how many steps there are in diffusion (normal diffusion + upsampler)
substep_count: 1 # remember to update in diffusion config in pointe, under the k keys
cond_config:
- name: "SoftZoo_Sim"
grad_scale: !!float 1e6
grad_clamp: !!float 1e0
calc_gradient: True
logging_bool: True
# - name: "Dist_To_Origin"
# grad_scale: !!float 1e0
# grad_clamp: !!float 1e-2
# calc_gradient: True
# logging_bool: True
cond_overall_logging: True
preload_emb:
condition_embedding:
shape: [1,3,224,224]
path: asset/img_asset/ver_rec.png
random: False
diffusion_noise:
shape: [2,6,1024]
path: asset/embeddings/base_noise_cone.npy
upsample_noise:
shape: [1,6,3072]
path: asset/embeddings/upsample_noise_cone.npy
softzoo_config:
custom_gravity: False
# eval: True
out_dir: null # Placeholder, need to be set in run.py
fixed_v: gripping_5_HTP.json
env_config_file: null # Placeholder, need to be inserted manually
save_every_iter: *log_freq
render_every_iter: *log_freq
n_iters: 301
n_frames: 100
controller_type: all_on
active: True
action_v_strength: 0.35
loss_types: [ObjectBalancedLoss,ContactLoss,DiffOrientationLoss]
loss_coefs: [1.0,1.0,0.1]
obj_x_mul: [0.,1.,0.]
obj_particle_id: 2
surface_threshold: 0.02
seed: *seed # Placeholder, do not use
torch_seed: *seed # Placeholder, do not use
static_as_fixed: True
device_memory_fraction: 0.7
optimize_designer: True
optimize_design_types: ['geometry','softness','suction_val']
designer_lr: 0.1
optimize_controller: True
gen_pointe_bounding_box:
max: [1.5,0.1,1.5]
# mean: [0.,0.,0.]
min: [-1.5,-3.5,-1.5]
design_device: cpu
obj_initial_pos: [0,0,0]
num_fingers: 2
pointe_config:
base_model_name: base40M
device: cuda:1
guidance_scale: [3.0, 0.0]
num_points: [1024, 3072]
sampling_mode: ddim
|