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

[Question] validation cross entropy #2000

Closed
YassineYousfi opened this issue Nov 30, 2020 · 11 comments
Closed

[Question] validation cross entropy #2000

YassineYousfi opened this issue Nov 30, 2020 · 11 comments

Comments

@YassineYousfi
Copy link
Collaborator

Is the 0.051 validation cross entropy up to date? Are there any FLOPs constraints?

@geohot
Copy link
Contributor

geohot commented Dec 1, 2020

Yes, and that was with a unet. But if you can do a lot better with more FLOPS, go for it!

@YassineYousfi
Copy link
Collaborator Author

Thanks!
I used a unet structure too with efficientnet-B4 as encoder, no zooming or TTA. Getting val CE = 0.045.
I can probably do better with weather/lighting augmentations.
I trained first on (H//2, W//2) then continued on full res. Training on full res from scratch gives me around 0.05x too.
Great dataset! I can't wait for the full release.

@geohot
Copy link
Contributor

geohot commented Dec 2, 2020

Sweet! Want to open source it?

@dorkmo
Copy link
Contributor

dorkmo commented Dec 2, 2020

@YassineYousfi
Copy link
Collaborator Author

Yes @dorkmo, @geohot I will work on it a bit more this weekend and release the weights too.

@dorkmo
Copy link
Contributor

dorkmo commented Dec 2, 2020

very cool! ill share with the #comma-pencil guys on discord

@geohot
Copy link
Contributor

geohot commented Dec 7, 2020

Great stuff! Can you add a LICENSE and release weights?

And I didn't know about "segmentation_models.pytorch" or "albumentations", pytorch is fire!

@geohot
Copy link
Contributor

geohot commented Dec 7, 2020

It's crazy that this doesn't overfit, you aren't scale augmenting I don't see.

Added link to our README with the new record.

@YassineYousfi
Copy link
Collaborator Author

The pytorch ecosystem is 🔥
Sure, I will release weights and a new version ASAP. (I'm slowed down by a conference I'm attending this week)
Thanks for linking to the repo!! I added a MIT license.

@YassineYousfi
Copy link
Collaborator Author

There is some regularization (weight_decay), but I would guess that the robustness to overfitting mostly comes from Imagenet pre-trained encoder and the augmentations.

@YassineYousfi
Copy link
Collaborator Author

I released the weights here: https://github.com/YassineYousfi/comma10k-baseline/releases/tag/v0.1-alpha

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