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

Add faster-rcnn lightning module #10

Open
wants to merge 5 commits into
base: master
Choose a base branch
from
Open

Add faster-rcnn lightning module #10

wants to merge 5 commits into from

Conversation

Kshitij09
Copy link
Owner

  • Added required configurations for the initial LightningModule implementations
  • Added FasterRCNN Module, currently using all the default parameters of model definition
  • Added minimal run script for training faster-rcnn module using hydra configurations

The training script runs with no errors for PL's fast-dev-run=True. Will be adding more rigorous tests in the future

@Kshitij09 Kshitij09 added the hacktoberfest-accepted Accept for hacktoberfest, will merge later label Oct 14, 2020
* Added required configurations for the initial LightningModule implementations
* Added FasterRCNN Module, currently using all the default parameters of model definition
* Added minimal run script for training faster-rcnn module using hydra configurations

Signed-off-by: Kshitij Patil <kshitijpatil98@gmail.com>
* Added config for trainer, loggers and callbacks
* Updated run script to support logging

Signed-off-by: Kshitij Patil <kshitijpatil98@gmail.com>

Format tests

Signed-off-by: Kshitij Patil <kshitijpatil98@gmail.com>
Signed-off-by: Kshitij Patil <kshitijpatil98@gmail.com>
Signed-off-by: Kshitij Patil <kshitijpatil98@gmail.com>
Signed-off-by: Kshitij Patil <kshitijpatil98@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
hacktoberfest-accepted Accept for hacktoberfest, will merge later
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant