RuntimeError: MPS: Unsupported Border padding mode

[bukhalmae145]

RuntimeError: MPS: Unsupported Border padding mode

[jonahclarsen]

I'm getting the same error when running https://github.com/google-research/frame-interpolation on my M2 MBA with torch 2.1.2:

  File "\torch\nn\functional.py", line 4244, in grid_sample
        align_corners = False

    return torch.grid_sampler(input, grid, mode_enum, padding_mode_enum, align_corners)
           ~~~~~~~~~~~~~~~~~~ <--- HERE
RuntimeError: MPS: Unsupported Border padding mode

[jonahclarsen]

I figured it out. The solution was, before exporting the torchscript model, change the padding mode from 'border' to 'zeros'. Haven't yet seen how this affects the model outputs however.

[bukhalmae145]

I figured it out. The solution was, before exporting the torchscript model, change the padding mode from 'border' to 'zeros'. Haven't yet seen how this affects the model outputs however.

get this error if I change to 'zeros'

File "/Users/workstation/Movies/AFI-ForwardDeduplicate/models/model_pg104/forward_warp2.py", line 173, in sample_one img_warp.put_(ids_mask, torch.masked_select(flat_img*flat_weight, mask), accumulate=True) IndexError: out of range: tried to access index 33423360 on a tensor of 33423360 elements.