Prediction task

I got this error when I am running my main code, I need your help"
Traceback (most recent call last):
File “C:\Users\solo\Desktop\EXPERIMENT\ML-master\main.py”, line 188, in
w, loss = local.train(net=copy.deepcopy(net_glob).to(args.device))
File “C:\Users\solo\Desktop\EXPERIMENT\ML-master\models\Update.py”, line 59, in train
loss = self.loss_func(log_probs, labels)
File “C:\Users\solo\AppData\Roaming\Python\Python310\site-packages\torch\nn\modules\module.py”, line 1130, in _call_impl
return forward_call(*input, **kwargs)
File “C:\Users\solo\AppData\Roaming\Python\Python310\site-packages\torch\nn\modules\loss.py”, line 1164, in forward
return F.cross_entropy(input, target, weight=self.weight,
File “C:\Users\solo\AppData\Roaming\Python\Python310\site-packages\torch\nn\functional.py”, line 3014, in cross_entropy
return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing)
RuntimeError: “nll_loss_forward_reduce_cuda_kernel_2d_index” not implemented for ‘Float’