[BUG] Object Detection AutoTrain Error: iteration over a 0-d tensor about autotrain-advanced

[abhishekkrthakur]

did you also upload validation data or just training data?

[rileybolen]

I only uploaded training data, it looked like it automatically did the train/val split. I did find an image that was listed in my metadata twice, so I am wondering if maybe it one of those entries ended up in validation and one in training, causing the image to not be found in the validation set. I fixed this and I am trying again. I can also try manually splitting and uploading my validation data. I will let you know if that fixes the error.

[abhishekkrthakur]

it does auto splitting. that shouldnt be an issue.

I did find an image that was listed in my metadata twice, so I am wondering if maybe it one of those entries ended up in validation and one in training, causing the image to not be found in the validation set. I fixed this and I am trying again

please let me know. this case should be caught earlier

[rileybolen]

@abhishekkrthakur I tried removing the duplicated image record from metadata.jsonl and I still got the same error.

[abhishekkrthakur]

okay. so the issue is happening for datasets that have a single class. im fixing the issue and will update here asap.
i really hope it works for you end to end now. and deep apologies.

[rileybolen]

@abhishekkrthakur Sounds good, thanks! And no problem, I'm glad I can help test a new feature.

[abhishekkrthakur]

just pushed a fix and tried on my own please make sure you are on v0.7.110 or above.

[abhishekkrthakur]

please let me know if you still face issues

[rileybolen]

It seems that the training has worked, thanks! I am just facing issues now with the Serverless Inference API, but I think that is separate from this repo. So I think this issue is solved now!

[abhishekkrthakur]

The api wont work immediately. Try a few minutes after training is done :) and thank you so much for all the help :)

[abhishekkrthakur]

@rileybolen thank you very much for helping debugging this and apologies for the inconveniences. As a gratitude, we have added a $25 credit to your hugging face account that you can use for spaces, inference endpoints, autotrain or other huggingface services.

[abhishekkrthakur]

fixed