train-yolos error

[roboflow]

train-yolos error

[SkalskiP]

Hello again @robmarkcole 👋🏻! I confirm that the train-yolos-huggingface-object-detection-on-custom-data.ipynb notebook does not work even with our own dataset right now. Judging by logs coming from transformers and pytorch-lightning most likely because of some pip package update. We will work on fixing that issue with the rest of the team.

[SkalskiP]

I just released a fix for train-yolos-huggingface-object-detection-on-custom-data.ipynb. I tried it on Google Colab and it works for me. Please take a look, and let me know if you stil experience issues you described.

[robmarkcole]

OK run the new notebook and seems OK. Just noted the change to accelerator='gpu', devices=1

[SkalskiP]

@i-m-karanagarwal, could you grant me access? I just requested :)

[SkalskiP]

Hi, @robmarkcole 👋🏻! Thank you very much for your interest in Roboflow Notebooks! Please have a little patience. I'm taking a look at your problem, but it may take some time to figure out what's happening. I'll keep you posted.

[robmarkcole]

@SkalskiP I just noticed batch_size=1 which is an odd choice

[SkalskiP]

I'm not sure why @Jacobsolawetz decided to use batch_size=1. Maybe problems with GPU memory overflow. Glad to hear that everything works.

[i-m-karanagarwal]

@SkalskiP I tried to run the notebook with my own dataset but it showed the same error.
Could you please check what's causing it?

[SkalskiP]

@i-m-karanagarwal just to be sure, could you send me a link to Google Colab that you are using?

[i-m-karanagarwal]

@SkalskiP Yes here it is https://colab.research.google.com/drive/1Xf6NmlHIs4YiemhDqEBA81eBossgNvWl#scrollTo=0_YBNFmWivR0

[SkalskiP]

Okay, I know what's going on! :) This is an old version of the notebook, before the fix. Please use this version: https://colab.research.google.com/github/roboflow-ai/notebooks/blob/main/notebooks/train-yolos-huggingface-object-detection-on-custom-data.ipynb

@i-m-karanagarwal, could you please let me know where did you found the link to that version? We need to update that link so that others wouldn't struggle with the same errors :)

[i-m-karanagarwal]

Okay okay
@SkalskiP I found the old version on the documentation site of Roboflow Under YoloS model.

[SkalskiP]

Thank you very much 🙏🏻 @i-m-karanagarwal! I'll make sure that this time the link gets updated :)