- # for t in range(frame_sequences.size(1)):
- # img = wireworld.seq2img(frame_sequences[:, t])
- # torchvision.utils.save_image(
- # img.float() / 255.0,
- # f"/tmp/frame_{t:03d}.png",
- # nrow=8,
- # padding=6,
- # pad_value=0,
- # )
+ for t in range(frame_sequences.size(1)):
+ img = wireworld.seq2img(frame_sequences[:, t])
+ torchvision.utils.save_image(
+ img.float() / 255.0,
+ f"/tmp/frame_{t:03d}.png",
+ nrow=8,
+ padding=6,
+ pad_value=0,
+ )