RuntimeError: grad can be implicitly created only for scalar outputs

[ming-ming15]

Hi, thanks for your amazing work.
The following error occurs when running the provided bash file:
Traceback (most recent call last):
File "/media/disk8T/zmx/sparseneus/exp_runner_finetune.py", line 609, in
runner.train()
File "/media/disk8T/zmx/sparseneus/exp_runner_finetune.py", line 320, in train
loss.backward()
File "/home/zhumingxia/anaconda3/envs/spneus/lib/python3.10/site-packages/torch/tensor.py", line 488, in backward
torch.autograd.backward(
File "/home/zhumingxia/anaconda3/envs/spneus/lib/python3.10/site-packages/torch/autograd/init.py", line 190, in backward
grad_tensors = make_grads(tensors, grad_tensors, is_grads_batched=False)
File "/home/zhumingxia/anaconda3/envs/spneus/lib/python3.10/site-packages/torch/autograd/init.py", line 85, in _make_grads
raise RuntimeError("grad can be implicitly created only for scalar outputs")
Thank you in advance.