e-packages/ttach/functional.py", line 7, in rot90 return torch.rot90(x, k, (2, 3)) RuntimeError: Rotation dim0 out of range, dim0 = 2 #13
Description
e-packages/ttach/functional.py", line 7, in rot90
return torch.rot90(x, k, (2, 3))
RuntimeError: Rotation dim0 out of range, dim0 = 2