Skip to content

Navigation Menu

Explore
By company size
By use case
By industry
View all solutions
Topics
- AI
- DevOps
- Security
- Software Development
- View all
Explore
- GitHub Sponsors
  Fund open source developers
- The ReadME Project
  GitHub community articles
Repositories
- Enterprise platform
  AI-powered developer platform
Available add-ons
Pricing

Search code, repositories, users, issues, pull requests...

Search

Clear

Search syntax tips

Provide feedback

We read every piece of feedback, and take your input very seriously.

Include my email address so I can be contacted

Saved searches

Use saved searches to filter your results more quickly

Name

Query

To see all available qualifiers, see our documentation.

Sign up

You signed in with another tab or window. Reload to refresh your session. You signed out in another tab or window. Reload to refresh your session. You switched accounts on another tab or window. Reload to refresh your session.

Dismiss alert

huggingface / transformers Public

Notifications You must be signed in to change notification settings
Fork 28.8k
Star 144k

Code
Issues 1.1k
Pull requests 716
Actions
Projects 1
Security
Insights

Additional navigation options

Code
Issues
Pull requests
Actions
Projects
Security
Insights

Enable users to use their own loss functions + deal with prefetching for grad accum#34198

Merged

muellerzr merged 20 commits intomainhuggingface/transformers:mainfrom

muellerzr-fix-loss-calchuggingface/transformers:muellerzr-fix-loss-calcCopy head branch name to clipboard

Oct 17, 2024

Conversation Commits20 (20)Checks Files changed

Commits

Commits on Oct 16, 2024

bookmark
muellerzr
committed
Bookmark
muellerzr
committed
Bookmark
muellerzr
committed
Actually implement
muellerzr
committed
Pass in kwarg explicitly
muellerzr
committed
Adjust for if we do or don't have labels
muellerzr
committed
Bookmark fix for od
muellerzr
committed
bookmark
muellerzr
committed
Fin
muellerzr
committed
closer
muellerzr
committed
Negate accelerate grad accum div
muellerzr
committed

Commits on Oct 17, 2024

Fixup not training long enough
muellerzr
committed
Add in compute_loss to take full model output
muellerzr
committed
Document
muellerzr
committed
compute_loss -> compute_loss_fn
muellerzr
committed
Add a test
muellerzr
committed
Refactor
muellerzr
committed
Refactor
muellerzr
committed
Uncomment tests
muellerzr
committed
Update tests/trainer/test_trainer.py

muellerzr
and
danielhanchen
authored

Footer

© 2025 GitHub, Inc.

Footer navigation

Terms
Privacy
Security
Status
Docs
Contact

You can’t perform that action at this time.