Skip to content

Kageshimasu/focal-loss-with-smoothing

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Focal Loss with Label Smoothing

Label Smoothing applied in Focal Loss
This code is based on the below papers.

How to use

criteria = FocalLossWithSmoothing(num_classes)

logits = model(inputs)
loss = criteria(logits, labels)

optim.zero_grad()
loss.backward()
optim.step()

Releases

No releases published

Packages

No packages published

Languages