我正在使用传输学习为斯坦福汽车数据集构建一个分类模型。我想要实现标号平滑来惩罚过度自信的预测和改进泛化。
TensorFlow在CrossEntropyLoss中有一个简单的关键字参数。有人为PyTorch构建了类似的功能,我可以用它进行即插即用吗?
发布于 2019-04-15 09:00:49
据我所知没有。
下面是两个PyTorch实现的示例:
attention-is-all-you-need-pytorch,重新实现Google的注意就是你所需要的纸https://stackoverflow.com/questions/55681502
复制相似问题