May I know if you find the solution to this error? I have exactly the same problem using Pytorch lightening and forecasting. I get the same segmentation error. Mine also works with one GPU.
May I know if you find the solution to this error? I have exactly the same problem using Pytorch lightening and forecasting. I get the same segmentation error. Mine also works with one GPU.