(frames, path, fps=10)
| 77 | |
| 78 | |
| 79 | def _save_mp4(frames, path, fps=10): |
| 80 | import imageio |
| 81 | |
| 82 | if not frames: |
| 83 | return False |
| 84 | imageio.mimsave(path, [np.asarray(f).astype(np.uint8) for f in frames], fps=fps) |
| 85 | return True |
| 86 | |
| 87 | |
| 88 | def render_task(task_name, *, steps, seed, out_dir, side_by_side): |