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

Landan/deeplabv3 ade20k example #1593

Merged
merged 22 commits into from
Oct 14, 2022

Conversation

Landanjs
Copy link
Contributor

@Landanjs Landanjs commented Oct 5, 2022

What does this PR do?

Contributes example script to train DeepLabv3+ on ADE20k.

Old WandB runs for the full 128 epochs and without any constant LR:

mIoU Time-to-Train (hours)
Baseline + dice 46.3 3.85
Mild 46.84 3.31
Medium 49.13 5.0
Hot 48.22 6.68

I plan to rerun with constant LR and the target number of epochs for each recipe

@Landanjs Landanjs requested a review from coryMosaicML October 5, 2022 01:26
@Landanjs Landanjs requested a review from knighton as a code owner October 5, 2022 01:26
@Landanjs
Copy link
Contributor Author

Updated description with previous WandB runs, will attempt new runs with constant LR and appropriate recipe number of epochs

Copy link
Contributor

@coryMosaicML coryMosaicML left a comment

Choose a reason for hiding this comment

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

Great work! A few minor comments.

Biggest things:
Maybe include a little info on how to install things, or which docker image to use
One of the params for the hot recipe was a bit off.

examples/segmentation/README.md Outdated Show resolved Hide resolved
examples/segmentation/train_deeplabv3_ade20k.py Outdated Show resolved Hide resolved
examples/segmentation/train_deeplabv3_ade20k.py Outdated Show resolved Hide resolved
examples/segmentation/README.md Show resolved Hide resolved
@Landanjs Landanjs merged commit 9404bc0 into mosaicml:dev Oct 14, 2022
@Landanjs Landanjs deleted the landan/deeplabv3_ade20k_example branch October 14, 2022 16:53
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.

4 participants