fix(vis-traj-pred): torch.torch.set_grad_enabled(False) is (+1 more)

[andrewwhitecdw]

Small fixes in traj_pred/vis_traj_pred.py:

fix: torch.torch.set_grad_enabled(False) is not a valid call

Fix: Replace:

torch.torch.set_grad_enabled(False)

with:

torch.set_grad_enabled(False)

fix: missing output directory creation before writing videos

Fix: Replace:

    prefix = 'multistep_' if args.multi_step else 'singlestep_'
    vid_name = f'out/vis_traj_pred/{cfg.id}/v{version}_{cp_name}/{args.split}/{prefix}seq_len_{seq_len}/sd{args.seed}_s{i}_%s.mp4'
    # save GT
    visualizer.save_animation_as_video(

with:

    prefix = 'multistep_' if args.multi_step else 'singlestep_'
    vid_name = f'out/vis_traj_pred/{cfg.id}/v{version}_{cp_name}/{args.split}/{prefix}seq_len_{seq_len}/sd{args.seed}_s{i}_%s.mp4'
    os.makedirs(os.path.dirname(vid_name % 'gt'), exist_ok=True)
    # save GT
    visualizer.save_animation_as_video(

Files changed

• traj_pred/vis_traj_pred.py