Skip to content

wangleiofficial/label-smoothing-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

e7ea5e8 · Nov 2, 2020

History

4 Commits
Nov 2, 2020
Nov 2, 2020
Nov 2, 2020

Repository files navigation

label smoothing PyTorch implementation

Requirements

pytorch > 1.0

Install

git clone https://github.com/wangleiofficial/label-smoothing-pytorch.git

Usage

import LabelSmoothingCrossEntropy
criterion = LabelSmoothingCrossEntropy(reduction='sum')
loss = criterion(preds, labels)

License

MIT

Citation

Müller, Rafael, Simon Kornblith, and Geoffrey E. Hinton. "When does label smoothing help?." Advances in Neural Information Processing Systems. 2019.

About

label smoothing PyTorch implementation

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages