- frame_start, frame_end, actions = sequence()
- torchvision.utils.save_image(frame_start, "world_start.png")
- torchvision.utils.save_image(frame_end, "world_end.png")
+ frames, actions = sequence(nb_steps=31,all_frames=True)
+ frames = torch.cat(frames,0)
+ print(f"{frames.size()=}")
+ torchvision.utils.save_image(frames, "seq.png", nrow=8)