pytorch label smoothing代码

pytorch label smoothing代码

import torch
import torch.nn as nn
from torch.autograd import Variable
import matplotlib.pyplot as plt
import numpy as np
 
 
class LabelSmoothing(nn.Module):
    # "Implement label smoothing."
 
    def __init__(self, size, smoothing=0.0):
 
        super(LabelSmoothing, self).__init__()
 
        self.criterion = nn.KLDivLoss(size_average=False)
 
        #self.padding_idx = padding_idx
 
        self.confidence = 1.0 - smoothing
 
        self.smoothing = smoothing
 
        self.size = size
 
        self.true_dist = None
 
    def forward(self