TSN/SDN>> label_smoothing_pytorch>> 返回
项目作者： lonePatient

项目描述 ：
pytorch implement of Label Smoothing
高级语言： Python
项目地址: git://github.com/lonePatient/label_smoothing_pytorch.git
创建时间： 2019-12-16T13:13:23Z
项目社区：https://github.com/lonePatient/label_smoothing_pytorch
开源协议：
下载


Label Smoothing Pytorch

This repository contains a PyTorch implementation of the Label Smoothing.

Dependencies

  • PyTorch
  • torchvision
  • matplotlib
  • scikit-learn

Example

To produce th result, we use CIFAR-10 dataset for ResNet18.

  1. # no label smoothing
  2. python run.py
  4. # use label smoothing
  5. python run.py --do_lsr
  7. # extract feature
  8. python generate_feature.py
  10. python generate_feature.py --do_lsr
  12. #----------- plot tsne
  13. python tsne_plot.py
  15. python tsne_plot.py --do_lsr

Results

Training result

TSNE Visualisation