Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Train custom dataset #1

Open
blueclowd opened this issue Jan 13, 2020 · 6 comments
Open

Train custom dataset #1

blueclowd opened this issue Jan 13, 2020 · 6 comments

Comments

@blueclowd
Copy link

Hi @ruizengalways

Thanks for sharing the code. May I know if there is any tutorial showing how to train a custom dataset apart from COCO? Thanks

@ruizengalways
Copy link
Owner

You can use next() to iterate the dataset generator to know the details inside. I think it is not too hard to extend the generator to other datasets

@SuperDetected
Copy link

Excuse me!
Thanks for your codes.When I use your codes to train in the dataset COCO, the loss is always at 11.xx, and the MACE is always at 24.xx.They never go down.Could you tell me if there are something to adjust?

@ruizengalways
Copy link
Owner

Excuse me!
Thanks for your codes.When I use your codes to train in the dataset COCO, the loss is always at 11.xx, and the MACE is always at 24.xx.They never go down.Could you tell me if there are something to adjust?

How many epochs have you already trained? Could you please give me more details about your training?

@SuperDetected
Copy link

SuperDetected commented May 27, 2020 via email

@ruizengalways
Copy link
Owner

Increase the batch size and decrease the learning rate

Thanks for your reply,I trained the dataset just with the epochs you have set in the codes: And the only thing I changed is the IMAGES_PER_GPU,the original value is 32 and I set it 2 like this:

------------------ 原始邮件 ------------------ 发件人: "Rui Zeng"<[email protected]>; 发送时间: 2020年5月27日(星期三) 上午10:03 收件人: "ruizengalways/PFNet"<[email protected]>; 抄送: "知足~奔跑"<[email protected]>;"Comment"<[email protected]>; 主题: Re: [ruizengalways/PFNet] Train custom dataset (#1) Excuse me! Thanks for your codes.When I use your codes to train in the dataset COCO, the loss is always at 11.xx, and the MACE is always at 24.xx.They never go down.Could you tell me if there are something to adjust? How many epochs have you already trained? Could you please give me more details about your training? — You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe.

@SuperDetected
Copy link

SuperDetected commented May 27, 2020 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants