UnLiteFlowNet-PIV
UnLiteFlowNet-PIV copied to clipboard
RuntimeError: The size of tensor a (4) must match the size of tensor b (12) at non-singleton dimension 0
Hi, i am getting the error RuntimeError: The size of tensor a (4) must match the size of tensor b (12) at non-singleton dimension 0 at this snippet EPE_map = torch.norm(target_flow - input_flow, 2, 1)
Any advice?