Ducking-WJ comments

Repositories
Issues
Comments

Results 1 comments of


                                            Ducking-WJ

sampling_locations = (ref + grid * offset_scale).repeat(N_, 1, 1, 1, 1).flatten(3, 4) + \ RuntimeError: The size of tensor a (16) must match the size of tensor b (32) at non-singleton dimension 2

I'm having a similar issue: sampling_locations = (ref + grid * offset_scale).repeat(N_, 1, 1, 1, 1).flatten(3, 4) + \ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: The size of tensor a (86) must match the...