mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-23 01:41:54 +00:00
quick fix
This commit is contained in:
@@ -199,7 +199,7 @@ def rollout(
|
|||||||
|
|
||||||
# Track the final observation.
|
# Track the final observation.
|
||||||
if return_observations:
|
if return_observations:
|
||||||
observation = preprocess_observation(observation, cfg=policy.config)
|
observation = preprocess_observation(observation)
|
||||||
all_observations.append(deepcopy(observation))
|
all_observations.append(deepcopy(observation))
|
||||||
|
|
||||||
# Stack the sequence along the first dimension so that we have (batch, sequence, *) tensors.
|
# Stack the sequence along the first dimension so that we have (batch, sequence, *) tensors.
|
||||||
|
|||||||
Reference in New Issue
Block a user