(output_params, output_dir)
| 32 | return res |
| 33 | |
| 34 | def save_params(output_params, output_dir): |
| 35 | # Convert to CPU Numpy Arrays |
| 36 | to_save = params2cpu(output_params) |
| 37 | # Save the Parameters containing the Gaussian Trajectories |
| 38 | os.makedirs(output_dir, exist_ok=True) |
| 39 | print(f"Saving parameters to: {output_dir}") |
| 40 | save_path = os.path.join(output_dir, "params.npz") |
| 41 | np.savez(save_path, **to_save) |
| 42 | |
| 43 | def save_params_ckpt(output_params, output_dir, time_idx): |
| 44 | # Convert to CPU Numpy Arrays |
no test coverage detected