Download config.yaml from DAVIAN-Robotics/DiscoDemo-Stage1_RL-StackCube-alpha0: direct link, hf CLI and curl.
- Browser
- Download file 2.98 kB
-
https://huggingface.co/DAVIAN-Robotics/DiscoDemo-Stage1_RL-StackCube-alpha0/resolve/main/config.yaml
- Command line
-
hf download hf://DAVIAN-Robotics/DiscoDemo-Stage1_RL-StackCube-alpha0/config.yaml
-
curl -L -o config.yaml https://huggingface.co/DAVIAN-Robotics/DiscoDemo-Stage1_RL-StackCube-alpha0/resolve/main/config.yaml
2.98 kB
| project_name: DiscoDemo | |
| entity_name: null | |
| group_name: stack_cube | |
| exp_name: prfcl-stack_cube | |
| seed: 0 | |
| logger_type: wandb | |
| save_path: libs/FlashSAC/models/stack_cube/prfcl-stack_cube/seed0-TIMESTAMP | |
| agent_load_path: null | |
| num_env_steps: 50000000 | |
| num_train_envs: 2048 | |
| updates_per_interaction_step: 64 | |
| gamma: 0.99 | |
| n_step: 8 | |
| env: | |
| workcell: libs/real2sim/assets/workcell.json | |
| task: libs/real2sim/real2sim/tasks/stack_cube.py | |
| env_version: fr3-stack_cube | |
| gripper_lock: null | |
| success_hold_steps: 20 | |
| agent: | |
| agent_type: flashSAC | |
| seed: 0 | |
| device_type: cuda | |
| buffer_max_length: 100000000 | |
| buffer_min_length: 5000 | |
| buffer_device_type: cuda | |
| sample_batch_size: 2048 | |
| normalize_reward: true | |
| normalized_G_max: 5.0 | |
| asymmetric_observation: false | |
| learning_rate_init: 0.0003 | |
| learning_rate_peak: 0.0003 | |
| learning_rate_end: 0.00015 | |
| learning_rate_warmup_rate: 1.0e-06 | |
| learning_rate_warmup_step: 50 | |
| learning_rate_decay_rate: 1.0 | |
| learning_rate_decay_step: 50000000 | |
| actor_num_blocks: 2 | |
| actor_hidden_dim: 128 | |
| actor_bc_alpha: 0.0 | |
| actor_noise_zeta_mu: 2.0 | |
| actor_noise_zeta_max: 16 | |
| actor_update_period: 2 | |
| critic_num_blocks: 2 | |
| critic_hidden_dim: 256 | |
| critic_num_bins: 101 | |
| critic_min_v: -5 | |
| critic_max_v: 5 | |
| critic_target_update_tau: 0.01 | |
| temp_initial_value: 0.01 | |
| temp_target_sigma: 0.15 | |
| temp_target_entropy: null | |
| gamma: 0.99 | |
| n_step: 8 | |
| use_compile: true | |
| compile_mode: auto | |
| use_amp: true | |
| load_optimizer: true | |
| load_reward_normalizer: true | |
| rfcl: | |
| demo_ratio: 0.5 | |
| truncate_after_success: 50 | |
| reverse: | |
| reverse_step_size: 48 | |
| advance_threshold: 0.9 | |
| frontier_window: 10 | |
| minimum_episode_steps: 30 | |
| solved_frac_threshold: 0.5 | |
| demo_path: data/demo_banks/stack_cube/human_demos.h5 | |
| init_pose: box:0.4125,0.6375,-0.15,0.15,45,0.15 | |
| eval_worker: | |
| num_envs: 50 | |
| skill: | |
| phi_hidden: | |
| - 256 | |
| - 256 | |
| phi_lr: 0.0001 | |
| dual_slack: 0.001 | |
| lam_init: 1.0 | |
| alpha: 0.0 | |
| z_dim: 0 | |
| safety_penalty: | |
| illegal_contact: | |
| weight: 0.0001 | |
| sat: 100 | |
| robot_bodies: | |
| - panda_link5 | |
| - panda_link6 | |
| - panda_link7 | |
| - panda_hand | |
| - panda_leftfinger | |
| - panda_rightfinger | |
| scene_bodies: | |
| - cube_red | |
| - cube_blue | |
| - table | |
| allowed_pairs: | |
| - - panda_leftfinger | |
| - cube_red | |
| - - panda_rightfinger | |
| - cube_red | |
| - - panda_leftfinger | |
| - cube_blue | |
| - - panda_rightfinger | |
| - cube_blue | |
| obs_bounds: | |
| object_z_m: | |
| - 0.0 | |
| - 0.6 | |
| eef_box_m: | |
| x: | |
| - 0.2 | |
| - 0.85 | |
| 'y': | |
| - -0.45 | |
| - 0.45 | |
| z: | |
| - 0.0 | |
| - 0.7 | |
| object_xy: | |
| - - 0.2 | |
| - 0.85 | |
| - - -0.4 | |
| - 0.4 | |
| task_name: stack_cube | |
| jerk_limited_action: | |
| max_joint_velocity: | |
| - 0.248734 | |
| - 0.680387 | |
| - 0.239029 | |
| - 0.694292 | |
| - 0.382944 | |
| - 0.623874 | |
| - 0.693818 | |
| max_joint_acceleration: | |
| - 0.504737 | |
| - 1.331577 | |
| - 0.521252 | |
| - 1.995616 | |
| - 1.534215 | |
| - 1.765087 | |
| - 2.173107 | |
| max_joint_jerk: | |
| - 3.050796 | |
| - 6.724498 | |
| - 3.02558 | |
| - 9.382696 | |
| - 11.665951 | |
| - 10.390659 | |
| - 15.262546 | |
| method_name: prfcl | |