mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-17 01:30:14 +00:00
Merge remote-tracking branch 'Cadene/user/rcadene/2024_03_31_remove_torchrl' into refactor_act_remove_torchrl
This commit is contained in:
@@ -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
|
||||||
Reference in New Issue
Block a user