Upload folder using huggingface_hub

.gitattributes

@@ -36,3 +36,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 train_state/ocdbt.process_0/d/2581b3cb54e8b88f2fca2fe4c55d2155 filter=lfs diff=lfs merge=lfs -text
 train_state/ocdbt.process_0/d/42df6e2221124eadd432d0dbd5519cd6 filter=lfs diff=lfs merge=lfs -text
 train_state/ocdbt.process_0/d/bfb28b30cc97892c31f438ded8c6ab6d filter=lfs diff=lfs merge=lfs -text
+train_state/ocdbt.process_0/d/67fcfe9dc0268a2959138c325a2caff1 filter=lfs diff=lfs merge=lfs -text
+train_state/ocdbt.process_0/d/a7cc8b1ccec1e73eecc8c3125c958e42 filter=lfs diff=lfs merge=lfs -text
+train_state/ocdbt.process_0/d/ec2aa8f4304ac1d2188f145437173243 filter=lfs diff=lfs merge=lfs -text

_CHECKPOINT_METADATA

@@ -1 +1 @@
-{"item_handlers": {"config": "orbax.checkpoint._src.handlers.json_checkpoint_handler.JsonCheckpointHandler", "metadata": "orbax.checkpoint._src.handlers.json_checkpoint_handler.JsonCheckpointHandler", "train_state": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1768580945479056382, "commit_timestamp_nsecs": 1768580946309179572, "custom_metadata": {}}
+{"item_handlers": {"config": "orbax.checkpoint._src.handlers.json_checkpoint_handler.JsonCheckpointHandler", "metadata": "orbax.checkpoint._src.handlers.json_checkpoint_handler.JsonCheckpointHandler", "train_state": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1768645757943165321, "commit_timestamp_nsecs": 1768645758782563716, "custom_metadata": {}}

config/metadata

@@ -1 +1 @@
-{"wandb": {"project": "musclemimic", "mode": "online", "tags": ["fullbody", "gmr", "energy"]}, "experiment": {"task_factory": {"name": "ImitationFactory", "params": {"amass_dataset_conf": {"dataset_group": "KIT_KINESIS_TRAINING_MOTIONS", "retargeting_method": "gmr", "gmr_config": {"src_human": "smplh", "target_fps": 30, "solver": "daqp", "damping": 0.5, "offset_to_ground": false, "use_velocity_limit": false, "use_fitted_shape": true, "shape_fitting_iterations": 500}}}}, "env_params": {"env_name": "MjxMyoFullBody", "headless": true, "horizon": 1000, "disable_fingers": true, "mjx_backend": "warp", "num_envs": 8192, "enable_muscle_length_observations": true, "enable_muscle_velocity_observations": true, "enable_muscle_force_observations": true, "enable_muscle_excitation_observations": true, "enable_muscle_activation_observations": true, "enable_touch_sensor_observations": true, "init_state_type": "TrajInitialStateHandler", "goal_type": "GoalTrajMimic", "goal_params": {"n_step_lookahead": 5, "n_step_stride": 20, "upper_body_xml_name": "torso", "enable_motion_phase": true, "use_concise_lookahead": true, "sites_for_mimic": ["pelvis_mimic", "upper_body_mimic", "head_mimic", "left_shoulder_mimic", "left_elbow_mimic", "left_hand_mimic", "right_shoulder_mimic", "right_elbow_mimic", "right_hand_mimic", "left_hip_mimic", "left_knee_mimic", "left_ankle_mimic", "left_toes_mimic", "right_hip_mimic", "right_knee_mimic", "right_ankle_mimic", "right_toes_mimic"]}, "control_type": "DefaultControl", "control_params": {}, "reward_type": "MimicReward", "reward_params": {"qpos_w_sum": 0.5, "qvel_w_sum": 0.1, "root_vel_w_sum": 0.1, "rpos_w_sum": 0.3, "rquat_w_sum": 0.01, "rvel_w_sum": 0.1, "use_mean_exp_reward": false, "sites_for_mimic": ["pelvis_mimic", "upper_body_mimic", "head_mimic", "left_shoulder_mimic", "left_elbow_mimic", "left_hand_mimic", "right_shoulder_mimic", "right_elbow_mimic", "right_hand_mimic", "left_hip_mimic", "left_knee_mimic", "left_ankle_mimic", "left_toes_mimic", "right_hip_mimic", "right_knee_mimic", "right_ankle_mimic", "right_toes_mimic"], "activation_energy_coeff": 0.1}, "terminal_state_type": "MeanRelativeSiteDeviationTerminalStateHandler", "terminal_state_params": {"mean_site_deviation_threshold": 0.3, "enable_site_check": true}}, "algorithm": "PPOJax", "actor_hidden_layers": [1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024], "critic_hidden_layers": [1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024], "activation": "silu", "lr": 0.0004, "num_envs": 8192, "total_timesteps": 4915200000, "max_grad_norm": 10, "anneal_lr": false, "lr_schedule_type": "linear", "warmup_steps": null, "min_lr_ratio": 0.1, "weight_decay": 0.001, "normalize_env": true, "debug": false, "optimizer_type": "muon", "muon_config": {"beta": 0.95, "nesterov": true, "ns_steps": 5, "weight_decay": 0.001, "adaptive": false}, "ppo_config": {"num_steps": 8, "update_epochs": 1, "num_minibatches": 8, "gamma": 0.99, "gae_lambda": 0.95, "clip_eps": 0.2, "clip_eps_vf": 0.2, "init_std": 1.0, "learnable_std": true, "ent_coef": 0.0, "vf_coef": 0.5}, "use_layernorm": true, "layernorm_eps": 1e-05, "use_moe": false, "moe_config": {"num_experts": 16, "moe_at_layers": [3], "apply_moe_to": "both", "apply_moe_to_output": false, "expert_hidden_ratio": 0.5, "temperature": 0.5, "load_balance_loss_weight": 0.01, "gate_init_std": 0.01}, "n_seeds": 1, "seeds": null, "vmap_across_seeds": false, "save_checkpoints": true, "max_checkpoints_to_keep": 10, "checkpoints_on_validation": true, "resume_from": "/users/ckli/musclemimic_dev/checkpoints/0ad10142e7af/checkpoint_38550", "async_checkpointing": false, "reset_logging_timestep": false, "reset_lr_schedule_on_resume": false, "auto_resume": true, "run_id": null, "checkpoint_root": null, "adaptive_termination": {"enabled": false, "init_threshold": 0.5, "low_band": 0.1, "high_band": 0.2, "adjust_factor": 0.95, "consecutive_k": 5, "min_threshold": 0.1, "ema_alpha": 0.1}, "reward_curriculum": {"enabled": false, "qvel_w_sum_init": 0.1, "root_vel_w_sum_init": 0.1, "qvel_w_sum_max": 0.4, "root_vel_w_sum_max": 0.4, "eta": 0.02, "ema_alpha": 0.2, "term_rate_threshold": 0.08, "consecutive_k": 5}, "validation": {"active": true, "deterministic": false, "num_steps": 1000, "num_envs": 20, "num": 500, "video_length": 3000, "video_frequency": 1, "video_fps": 60, "quantities": ["JointPosition", "JointVelocity", "RelSitePosition", "RelSiteVelocity", "RelSiteOrientation", "BodyPosition"], "measures": ["EuclideanDistance", "DynamicTimeWarping", "DiscreteFrechetDistance"], "rel_site_names": ["pelvis_mimic", "upper_body_mimic", "head_mimic", "left_shoulder_mimic", "left_elbow_mimic", "left_hand_mimic", "right_shoulder_mimic", "right_elbow_mimic", "right_hand_mimic", "left_hip_mimic", "left_knee_mimic", "left_ankle_mimic", "left_toes_mimic", "right_hip_mimic", "right_knee_mimic", "right_ankle_mimic", "right_toes_mimic"]}, "num_steps": 8, "update_epochs": 1, "num_minibatches": 8, "gamma": 0.99, "gae_lambda": 0.95, "clip_eps": 0.2, "clip_eps_vf": 0.2, "init_std": 1.0, "learnable_std": true, "ent_coef": 0.0, "vf_coef": 0.5, "num_updates": 75000, "minibatch_size": 8192, "validation_interval": 150, "effective_utd": 0.0001220703125, "sample_reuse": 1.0, "unrolls_per_1m": 15.2587890625, "updates_per_1m": 15.2587890625, "grad_steps_per_1m": 122.0703125, "checkpoint_dir": "/users/ckli/musclemimic_dev/checkpoints/0ad10142e7af", "resume_lr_override": 0.00039999998989515007}, "control_config": {"torque_control": {"control_type": "DefaultControl", "control_params": {}}}}
+{"wandb": {"project": "musclemimic", "mode": "online", "tags": ["fullbody", "gmr", "energy"]}, "experiment": {"task_factory": {"name": "ImitationFactory", "params": {"amass_dataset_conf": {"dataset_group": "KIT_KINESIS_TRAINING_MOTIONS", "retargeting_method": "gmr", "gmr_config": {"src_human": "smplh", "target_fps": 30, "solver": "daqp", "damping": 0.5, "offset_to_ground": false, "use_velocity_limit": false, "use_fitted_shape": true, "shape_fitting_iterations": 500}}}}, "env_params": {"env_name": "MjxMyoFullBody", "headless": true, "horizon": 1000, "disable_fingers": true, "mjx_backend": "warp", "num_envs": 8192, "enable_muscle_length_observations": true, "enable_muscle_velocity_observations": true, "enable_muscle_force_observations": true, "enable_muscle_excitation_observations": true, "enable_muscle_activation_observations": true, "enable_touch_sensor_observations": true, "init_state_type": "TrajInitialStateHandler", "goal_type": "GoalTrajMimic", "goal_params": {"n_step_lookahead": 5, "n_step_stride": 20, "upper_body_xml_name": "torso", "enable_motion_phase": true, "use_concise_lookahead": true, "sites_for_mimic": ["pelvis_mimic", "upper_body_mimic", "head_mimic", "left_shoulder_mimic", "left_elbow_mimic", "left_hand_mimic", "right_shoulder_mimic", "right_elbow_mimic", "right_hand_mimic", "left_hip_mimic", "left_knee_mimic", "left_ankle_mimic", "left_toes_mimic", "right_hip_mimic", "right_knee_mimic", "right_ankle_mimic", "right_toes_mimic"]}, "control_type": "DefaultControl", "control_params": {}, "reward_type": "MimicReward", "reward_params": {"qpos_w_sum": 0.5, "qvel_w_sum": 0.1, "root_vel_w_sum": 0.1, "rpos_w_sum": 0.3, "rquat_w_sum": 0.01, "rvel_w_sum": 0.1, "use_mean_exp_reward": false, "sites_for_mimic": ["pelvis_mimic", "upper_body_mimic", "head_mimic", "left_shoulder_mimic", "left_elbow_mimic", "left_hand_mimic", "right_shoulder_mimic", "right_elbow_mimic", "right_hand_mimic", "left_hip_mimic", "left_knee_mimic", "left_ankle_mimic", "left_toes_mimic", "right_hip_mimic", "right_knee_mimic", "right_ankle_mimic", "right_toes_mimic"], "activation_energy_coeff": 0.1}, "terminal_state_type": "MeanRelativeSiteDeviationTerminalStateHandler", "terminal_state_params": {"mean_site_deviation_threshold": 0.3, "enable_site_check": true}}, "algorithm": "PPOJax", "actor_hidden_layers": [1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024], "critic_hidden_layers": [1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024, 1024], "activation": "silu", "lr": 0.0004, "num_envs": 8192, "total_timesteps": 4915200000, "max_grad_norm": 10, "anneal_lr": false, "lr_schedule_type": "linear", "warmup_steps": null, "min_lr_ratio": 0.1, "weight_decay": 0.001, "normalize_env": true, "debug": false, "optimizer_type": "muon", "muon_config": {"beta": 0.95, "nesterov": true, "ns_steps": 5, "weight_decay": 0.001, "adaptive": false}, "ppo_config": {"num_steps": 8, "update_epochs": 1, "num_minibatches": 8, "gamma": 0.99, "gae_lambda": 0.95, "clip_eps": 0.2, "clip_eps_vf": 0.2, "init_std": 1.0, "learnable_std": true, "ent_coef": 0.0, "vf_coef": 0.5}, "use_layernorm": true, "layernorm_eps": 1e-05, "use_moe": false, "moe_config": {"num_experts": 16, "moe_at_layers": [3], "apply_moe_to": "both", "apply_moe_to_output": false, "expert_hidden_ratio": 0.5, "temperature": 0.5, "load_balance_loss_weight": 0.01, "gate_init_std": 0.01}, "n_seeds": 1, "seeds": null, "vmap_across_seeds": false, "save_checkpoints": true, "max_checkpoints_to_keep": 10, "checkpoints_on_validation": true, "resume_from": "/users/ckli/musclemimic_dev/checkpoints/0ad10142e7af/checkpoint_53550", "async_checkpointing": false, "reset_logging_timestep": false, "reset_lr_schedule_on_resume": false, "auto_resume": true, "run_id": null, "checkpoint_root": null, "adaptive_termination": {"enabled": false, "init_threshold": 0.5, "low_band": 0.1, "high_band": 0.2, "adjust_factor": 0.95, "consecutive_k": 5, "min_threshold": 0.1, "ema_alpha": 0.1}, "reward_curriculum": {"enabled": false, "qvel_w_sum_init": 0.1, "root_vel_w_sum_init": 0.1, "qvel_w_sum_max": 0.4, "root_vel_w_sum_max": 0.4, "eta": 0.02, "ema_alpha": 0.2, "term_rate_threshold": 0.08, "consecutive_k": 5}, "validation": {"active": true, "deterministic": false, "num_steps": 1000, "num_envs": 20, "num": 500, "video_length": 3000, "video_frequency": 1, "video_fps": 60, "quantities": ["JointPosition", "JointVelocity", "RelSitePosition", "RelSiteVelocity", "RelSiteOrientation", "BodyPosition"], "measures": ["EuclideanDistance", "DynamicTimeWarping", "DiscreteFrechetDistance"], "rel_site_names": ["pelvis_mimic", "upper_body_mimic", "head_mimic", "left_shoulder_mimic", "left_elbow_mimic", "left_hand_mimic", "right_shoulder_mimic", "right_elbow_mimic", "right_hand_mimic", "left_hip_mimic", "left_knee_mimic", "left_ankle_mimic", "left_toes_mimic", "right_hip_mimic", "right_knee_mimic", "right_ankle_mimic", "right_toes_mimic"]}, "num_steps": 8, "update_epochs": 1, "num_minibatches": 8, "gamma": 0.99, "gae_lambda": 0.95, "clip_eps": 0.2, "clip_eps_vf": 0.2, "init_std": 1.0, "learnable_std": true, "ent_coef": 0.0, "vf_coef": 0.5, "num_updates": 75000, "minibatch_size": 8192, "validation_interval": 150, "effective_utd": 0.0001220703125, "sample_reuse": 1.0, "unrolls_per_1m": 15.2587890625, "updates_per_1m": 15.2587890625, "grad_steps_per_1m": 122.0703125, "checkpoint_dir": "/users/ckli/musclemimic_dev/checkpoints/0ad10142e7af", "resume_lr_override": 0.00039999998989515007}, "control_config": {"torque_control": {"control_type": "DefaultControl", "control_params": {}}}}

metadata/metadata

@@ -1 +1 @@
-{"step": 369600, "update_number": 46200, "global_timestep": 3027763200, "learning_rate": 0.00039999998989515007, "num_envs": 8192, "num_steps": 8, "num_minibatches": 8, "update_epochs": 1, "schema_version": "2.1", "algo_version": "PPOJax_v1", "backend": "warp", "env_name": "NormalizeVecReward"}
+{"step": 433200, "update_number": 54150, "global_timestep": 3548774400, "learning_rate": 0.00039999998989515007, "num_envs": 8192, "num_steps": 8, "num_minibatches": 8, "update_epochs": 1, "schema_version": "2.1", "algo_version": "PPOJax_v1", "backend": "warp", "env_name": "NormalizeVecReward"}

train_state/ocdbt.process_0/d/67fcfe9dc0268a2959138c325a2caff1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0daf9f55daa250dcd2c3c2ae90d05298b9c28f67511e98999d4cb6332a2d6d40
+size 13119488

train_state/ocdbt.process_0/d/a7cc8b1ccec1e73eecc8c3125c958e42

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8264cfd6f065f53eb73be3456dd600da2271dbc226c93c590bc70a8cab7501f5
+size 124709

train_state/ocdbt.process_0/d/ec2aa8f4304ac1d2188f145437173243

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f1babc7cc921420241bfae16302999c00d6137a973e33bd02088b1d292f33f44
+size 261656576