While running the final prediction step of the pipeline the following warning is displayed
util.py:230: UserWarning: Using torch.cross without specifying the dim arg is deprecated.
Please either pass the dim explicitly or simply use torch.linalg.cross.
The default value of dim will change to agree with that of linalg.cross in a future release. (Triggered internally at ../aten/src/ATen/native/Cross.cpp:63.)
Z = torch.cross(Xn,Yn)
While running the final prediction step of the pipeline the following warning is displayed