Error

When I run the training/final step, I get this error can you advise?
^CTraceback (most recent call last):
File "train.py", line 198, in
main(args, configs)
File "train.py", line 93, in main
nn.utils.clip_grad_norm_(model.parameters(), grad_clip_thresh)
File "/home/layan/.local/lib/python3.6/site-packages/torch/nn/utils/clip_grad.py", line 36, in clip_grad_norm_
total_norm = torch.norm(torch.stack([torch.norm(p.grad.detach(), norm_type).to(device) for p in parameters]), norm_type)
File "/home/layan/.local/lib/python3.6/site-packages/torch/nn/utils/clip_grad.py", line 36, in
total_norm = torch.norm(torch.stack([torch.norm(p.grad.detach(), norm_type).to(device) for p in parameters]), norm_type)
File "/home/layan/.local/lib/python3.6/site-packages/torch/functional.py", line 1293, in norm
return _VF.norm(input, p, dim=_dim, keepdim=keepdim) # type: ignore
File "/home/layan/.local/lib/python3.6/site-packages/torch/_VF.py", line 25, in __getattr__
def __getattr__(self, attr):


Start with train.py at lines 93 and 198 and reproduce the error from the training/final step, using the traceback and linked screenshot as the available evidence. The issue does not specify a requested code change or expected behavior, so completion criteria are not defined.

Written by the indexing model from the issue text.