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

Make StochasticDepth FX-compatible #4373

Merged
merged 1 commit into from
Sep 6, 2021
Merged

Conversation

fmassa
Copy link
Member

@fmassa fmassa commented Sep 6, 2021

Follow-up from #4372

Let's just entirely wrap stochastic_depth, we can consider it as an elementary operator

Copy link
Contributor

@datumbox datumbox left a comment

Choose a reason for hiding this comment

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

LGTM, thanks!

@fmassa fmassa merged commit fbd69f1 into pytorch:main Sep 6, 2021
@fmassa fmassa deleted the fx-stochastic-depth branch September 6, 2021 19:08
facebook-github-bot pushed a commit that referenced this pull request Sep 9, 2021
Reviewed By: fmassa

Differential Revision: D30793328

fbshipit-source-id: 0659f255ce09f85e20abb217be8fb071a387537a
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants