data
GenieRolloutOutput
extends
ModelOutputGenieRolloutOutput(frames: Tensor, tokens: Tensor)Frames generated by playing the learned environment.
Attributes
Examples
>>> import lucid
>>> from lucid.models.generative.genie import GenieRolloutOutput
>>> out = GenieRolloutOutput(
... frames=lucid.zeros(1, 2, 3, 8, 8),
... tokens=lucid.zeros(1, 3, 4).to(lucid.int64))
>>> out.frames.shape
(1, 2, 3, 8, 8)