Views
No views yet
1s2_candidates_num=5
2noise_temp_lower_bound=1.0
3noise_temp_upper_bound=1.0
4time_temp_lower_bound=0.9
5time_temp_upper_bound=1.01from hume import HumePolicy
2import numpy as np
3
4# load policy
5hume = HumePolicy.from_pretrained("/path/to/checkpoints")
6
7# config Test-Time Computing args
8hume.init_infer(
9 infer_cfg=dict(
10 replan_steps=8,
11 s2_replan_steps=16,
12 s2_candidates_num=5,
13 noise_temp_lower_bound=1.0,
14 noise_temp_upper_bound=1.0,
15 time_temp_lower_bound=0.9,
16 time_temp_upper_bound=1.0,
17 post_process_action=True,
18 device="cuda",
19 )
20)
21
22# prepare observations
23observation = {
24 "observation.images.image": np.zeros((1,224,224,3), dtype = np.uint8), # (B, H, W, C)
25 "observation.images.wrist_image": np.zeros((1,224,224,3), dtype = np.uint8), # (B, H, W, C)
26 "observation.state": np.zeros((1, 7)), # (B, state_dim)
27 "task": ["Lift the papper"],
28}
29
30# Infer the action
31action = hume.infer(observation) # (B, action_dim)
321# source ckpts
22025-05-02/08-10-44_libero_goal_ck8-16-1_sh-4_gpu8_lr5e-5_1e-5_1e-5_2e-5_bs16_s1600k/0090000
3# original logs
42025-06-12/20-53-09+08-10-44_libero_goal_ck8-16-1_sh-4_gpu8_lr5e-5_1e-5_1e-5_2e-5_bs16_s1600k_0090000_s1-8_s2-16_s2cand-5_ntl-1.0_ntu-1.0_ttl-0.9_ttu-1.0.log1@article{song2025hume,
2 title={Hume: Introducing System-2 Thinking in Visual-Language-Action Model},
3 author={Anonimous Authors},
4 journal={arXiv preprint arXiv:2505.21432},
5 year={2025}
6}