Closed robmarkcole closed 1 year ago
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.
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.
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.
OK run the new notebook and seems OK. Just noted the change to accelerator='gpu', devices=1
@SkalskiP I just noticed batch_size=1
which is an odd choice
I'm not sure why @Jacobsolawetz decided to use batch_size=1
. Maybe problems with GPU memory overflow. Glad to hear that everything works.
@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?
@i-m-karanagarwal just to be sure, could you send me a link to Google Colab that you are using?
@SkalskiP Yes here it is https://colab.research.google.com/drive/1Xf6NmlHIs4YiemhDqEBA81eBossgNvWl#scrollTo=0_YBNFmWivR0
@i-m-karanagarwal, could you grant me access? I just requested :)
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 :)
Okay okay @SkalskiP I found the old version on the documentation site of Roboflow Under YoloS model.
Thank you very much 🙏🏻 @i-m-karanagarwal! I'll make sure that this time the link gets updated :)
Search before asking
Notebook name
train-yolos-huggingface-object-detection-on-custom-data.ipynb
Bug
I am using my own dataset from roboflow, exported in coco format.
At:
I get error:
Environment
Colab
Minimal Reproducible Example
No response
Additional
No response
Are you willing to submit a PR?