wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

WIP

This commit is contained in:
Remi Cadene 2024-09-18 00:36:08 +02:00
parent 850e6d540d
commit 568394ec06
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
test.py
View File

@ -45,9 +45,8 @@ class Policy:
latest_observation, latest_timestamp = self.obs_queue.get() latest_observation, latest_timestamp = self.obs_queue.get()
if previous_timestamp is not None and previous_timestamp == latest_timestamp: if previous_timestamp is not None and previous_timestamp == latest_timestamp:
time.sleep( # in case inference ran faster than recording/adding a new observation in the queue
0.1 time.sleep(0.1)
) # in case inference ran faster than recording/adding a new observation in the queue
else: else:
predicted_action_sequence = self.inference(latest_observation) predicted_action_sequence = self.inference(latest_observation)
self.action_queue.add(predicted_action_sequence, latest_timestamp) self.action_queue.add(predicted_action_sequence, latest_timestamp)