I am trying to train YOLOX algorithm on gaudi2. I am getting above error at this operation. grid = torch.stack((xv, yv), 2).view(1, 1, hsize, wsize, 2).type(dtype) How can I solve it? When posting a technical issue, please describe the issue; be as descriptive as possible, you can include things …

We support DDP for scaling in pytorch not horovod. Horovod is supported only on tensorflow. See here and here .

ValueError: invalid type: 'torch.hpu.FloatTensor'

Sayantan_S June 1, 2023, 5:31pm 8

From this post, it seems you have been able to run the yolox model?

Topic		Replies	Views
RuntimeError: [Rank:0] FATAL ERROR :: MODULE:PT_BRIDGE Exception in Lowering thread PyTorch	3	207	March 23, 2025
AttributeError : 'HabanaParameterWrapper' object has no attribute 'change_device_placement' Training pytorch	6	113	October 23, 2024
Gaudi1 HPU doesn't support long? PyTorch pytorch	11	305	April 4, 2024
RuntimeError: Input sizes must be equal when doing loss.backward() during the training of a GNN PyTorch pytorch	3	79	March 20, 2025
AttributeError : 'HabanaParameterWrapper' object has no attribute 'change_device_placement' PyTorch pytorch	1	56	September 24, 2024