File size: 1,880 Bytes
4b370b2
 
 
 
3526576
4b370b2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
a297347
47e3f60
3526576
 
4b370b2
3526576
4b370b2
 
 
 
 
 
 
3526576
a297347
4b370b2
a297347
4b370b2
 
617226e
 
61eb7c2
4b370b2
 
 
47e3f60
4b370b2
 
48c4a76
4b370b2
 
 
3db6c1b
4b370b2
a297347
4b370b2
 
 
 
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
exp_name: debug_server
seed: &seed 22
out_dir: ./logs/
tensorboard_log_dir: runs/
save_every_iter: &log_freq 12
total_steps: 512 # 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

softzoo_config:
  custom_gravity: False
  # eval: True
  out_dir: null # Placeholder, need to be set in run.py
  fixed_v: gripping_5_HTP.json
  objects_set_file: objects_set_a.json
  env_config_file: debug_server.yaml  # Placeholder, need to be inserted manually
  layout_config_file: double.json
  desc_prompt: base.txt
  viewpoint: [-1.,-0.25,-0.1]
  save_every_iter: *log_freq
  render_every_iter: 8
  n_iters: 301
  n_frames: 100

  controller_type: all_on
  active: True
  action_v_strength: 0.35

  loss_types: [ObjectBalancedLoss,ContactLoss,ThrowingObjectLoss]
  loss_coefs: [1.e-4,1.e-5,1.0]
  obj_x_mul: [0.,1.,0.]
  com_x_mul: [1.,2.,1.]
  obj_particle_id: 2

  design_loss_types:  ['FingerPenLoss','SelfFingerPenLoss','CLIPLoss','AngleLoss']
  design_loss_coeffs: [1.0,1.0,1.e-4,0.1]

  seed: *seed  # Placeholder, do not use
  torch_seed: *seed  # Placeholder, do not use
  static_as_fixed: True
  device_memory_fraction: 0.75

  optimize_designer: True
  optimize_design_types: ['geometry','softness_E','softness_v','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.0,-1.5]

  design_device: cpu
  obj_initial_pos: [0,0,0]
  num_fingers: 2