Giter Club home page Giter Club logo

dlb-pytorch's Introduction

DLB-Pytorch

A minimalist unofficial implementation of "Self-Distillation from the Last Mini-Batch for Consistency Regularization"

reference: https://arxiv.org/pdf/2203.16172.pdf

Prerequisites

  • PyTorch
  • PyTorch Lightning
  • timm
  • torchmetrics
  • torchvision
  • python3
  • CUDA

Citations

@misc{https://doi.org/10.48550/arxiv.2203.16172,
  doi = {10.48550/ARXIV.2203.16172},
  url = {https://arxiv.org/abs/2203.16172},
  author = {Shen, Yiqing and Xu, Liwu and Yang, Yuzhe and Li, Yaqian and Guo, Yandong},
  keywords = {Computer Vision and Pattern Recognition (cs.CV), FOS: Computer and information sciences, FOS: Computer and information sciences},
  title = {Self-Distillation from the Last Mini-Batch for Consistency Regularization},
  publisher = {arXiv},
  year = {2022},
  copyright = {Creative Commons Attribution 4.0 International}
}

dlb-pytorch's People

Contributors

dongkyunk avatar

Stargazers

Ryan avatar Jinsoo avatar  avatar An-zhi WANG avatar  avatar  avatar Guo Xun avatar Derrick avatar

Watchers

 avatar

dlb-pytorch's Issues

蒸馏损失函数问题

你好,感谢您的代码分享,非常棒的工作!

在将DLB迁移到我的工作时,我发现损失函数好像一些问题。
dlb_model.py中的第45行代码,使用了交叉熵损失来计算前后batch之间的损失,但是交叉熵损失传入的第二个参数是一个软标签,导致运行错误。
请问这里是不是有什么问题呢?

非常期待您的回复~

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.