chao.wu/Lerobot
1
0
Fork 0
Code Pull Requests Activity

policies/utils.py

Browse Source
This commit is contained in:
Cadene 2024-04-05 11:47:15 +00:00
parent a420714ee4
commit c17dffe944
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lerobot/common/policies/utils.py Normal file
View File

@ -0,0 +1,10 @@
def populate_queues(queues, batch):
for key in batch:
if len(queues[key]) != queues[key].maxlen:
# initialize by copying the first observation several times until the queue is full
while len(queues[key]) != queues[key].maxlen:
queues[key].append(batch[key])
else:
# add latest observation to the queue
queues[key].append(batch[key])
return queues