FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

Dropout Fix - Final Polishing: updated submissions, tests, docs. by Niccolo-Ajroldi · Pull Request #884 · mlcommons/algorithmic-efficiency · GitHub

Dropout Fix - Final Polishing: updated submissions, tests, docs. - #884

Merged
priyakasimbeg merged 9 commits into
mlcommons:devfrom
Niccolo-Ajroldi:dropout_subs_fix
Aug 26, 2025
Merged

Dropout Fix - Final Polishing: updated submissions, tests, docs.#884
priyakasimbeg merged 9 commits into
mlcommons:devfrom
Niccolo-Ajroldi:dropout_subs_fix

Conversation

Copy link
Copy Markdown
Member

The new dropout design in AlgoPerf (see #753 (comment) and #873) requires passing dropout_rate to the workload’s forward pass function rather than to init_model_fn. Some minor issues from this change were left unresolved, and this PR fixes them.

  • Reference submissions: we pipe dropout_rate through the model fwd pass in both JAX and PyTorch submissions.
  • PyTorch dev workloads: we adapt MNIST and CIFAR workloads to new dropout design.
  • modeldiff: we remove dropout_rate and aux_dropout_rate from tests/modeldiffs/diff.py and adapt all tests by pipping dropout_rate=0.0 through the test call (one-line change).
  • spec.py: updated spec to reflect new design.
  • DOCS: updated docs to reflect new design.

Niccolo-Ajroldi requested a review from a team as a code owner August 25, 2025 13:44

github-actions Bot commented Aug 25, 2025
edited
Loading

Copy link
Copy Markdown

MLCommons CLA bot All contributors have signed the MLCommons CLA ✍️ ✅

priyakasimbeg left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Choose a reason Spam Abuse Off Topic Outdated Duplicate Resolved Low Quality

LGTM! Thank you for tying up the loose ends Niccolo!

priyakasimbeg merged commit 0643268 into mlcommons:dev Aug 26, 2025
15 checks passed
github-actions Bot locked and limited conversation to collaborators Aug 26, 2025
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants


Back | FazBrowse Home | New Git URL