RuntimeError: grad can be implicitly created only for scalar outputs

[ditarnnt]

hi i tried the repo, i use yours datasets for training. i got error
RuntimeError: grad can be implicitly created only for scalar outputs

i tried to check the bug by Make sure the number of classes matches the annotation and the model architecture., where do i can find the classes matched the annotation? is it how many label you use in anno-converted.json?