sarthak1996:

RuntimeError: only batches of spatial targets supported (3D tensors) but got targets of dimension: 4


I’m seeing exactly the same error with @bodokaiser code (piwise). Has anyone found a solution?
Thanks