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

Does this model support the segmentation file? #11

Closed
Epiphqny opened this issue Dec 11, 2020 · 4 comments
Closed

Does this model support the segmentation file? #11

Epiphqny opened this issue Dec 11, 2020 · 4 comments

Comments

@Epiphqny
Copy link

No description provided.

@jackroos
Copy link
Member

Hi @Epiphqny

It is from the original DETR code and we have not tested it or train the segmentation model. But I think it is not a big problem to train the segmentation model following the instructions in DETR repo.

@Epiphqny
Copy link
Author

@jackroos Thanks for your quick reply, I will have a try.

@Epiphqny
Copy link
Author

@jackroos How should I adjust the learning rate if I want to use larger batch size? Do you have any advice?

@jackroos
Copy link
Member

@Epiphqny In my experience, when adam is used, you should keep the learning rate unchanged if the batch size does not change a lot. But for a much larger batch size, you may need to tune the learning rate by yourself.

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

2 participants