604e535
1
2
3
4
5
6
7
8
"""FlowMo rollout interface.""" def rollout(model, batch, horizon): """Run open-loop FlowMo prediction.""" images, actions, future_actions = batch return model.rollout(images, actions, future_actions[:, :horizon])
