Skip to content

Conversation

yoshitomo-matsubara
Copy link
Contributor

As discussed in #1606 the default hyperparameters in detection/train.py are different from those used to train Faster/Mask/Keypoint R-CNNs.

Following a PR suggestion from @fmassa, I updated the default hyperparameters, and added some comments so that we can have a better idea as we train them by ourselves e.g., using different datasets, models, etc.

Thank you!

Copy link
Member

@fmassa fmassa left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks a lot!

Can you also create a README in this folder in a follow-up PR with commands used to run the code, similar to https://github.com/pytorch/vision/tree/master/references/classification?

@yoshitomo-matsubara
Copy link
Contributor Author

Sure, I sent a follow-up PR #1612 :)

fmassa pushed a commit to fmassa/vision-1 that referenced this pull request Dec 18, 2019
facebook-github-bot pushed a commit that referenced this pull request Dec 19, 2019
Summary: Pull Request resolved: #1684

Reviewed By: cpuhrsch

Differential Revision: D19156997

Pulled By: fmassa

fbshipit-source-id: dca753d7e1e9041a9973f0c45a8046358e687188
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

Successfully merging this pull request may close these issues.

2 participants