GithubHelp home page GithubHelp logo

winycg / hsakd Goto Github PK

View Code? Open in Web Editor NEW
60.0 1.0 13.0 1.21 MB

[IJCAI-2021&&TNNLS-2022] Official implementation of Hierarchical Self-supervised Augmented Knowledge Distillation

License: Apache License 2.0

Python 96.74% Shell 3.26%

hsakd's People

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar

hsakd's Issues

About training time.

Hi, thanks for your wonderful work. We wonder about the detailed training GPU device and training time. And we train the model on 6 GeForce 3090 for now, it will take about 2 hours per epoch.

The meanings of ss_acc

I found the ss_acc1 and class_acc1 in your output .I want to ask a question the meanings of ss_acc1 ?

self-supervised lable construction

Thanks for your work. I want to know how to use self-supervised learning in this code. Do you implement self-supervised leanring through following code?

labels = torch.stack([target*4+i for i in range(4)], 1).view(-1)

Howerver, I think it is supervised learning label in above code?

About the model size

Hi, thanks for your cool work!

The released resnet18 checkpoint ' resnet18_imagenet_aux_best.pth.tar' has a size of about 452M, which is much larger than the original one (44.7M). What is the difference between them?

Thanks!

Size mismatch

Thanks for the great work! However, I found a mismatch error when importing your pre-trained weights(ResNet50_aux.pth.tar):

RuntimeError: Error(s) in loading state_dict for ResNet_Auxiliary:
        size mismatch for backbone.conv1.weight: copying a param with shape torch.Size([64, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([64, 3, 7, 7]).

Is it caused by the difference between the size of the convolution and the pre-trained model?

# models/resnet_imagenet.py   line32
self.conv1 = nn.Conv2d(3, self.inplanes, kernel_size=7, stride=2, padding=3, bias=False)

How can I go about solving it? Looking forward to your reply!

Reproduce other KD methods on Res50->MobileNetV2

I reproduced your method on Res50->MobileNetV2 and got 79.24 top-1 acc, that's awesome. But I reproduce other KD methods SSKD 、CRD... ,I found that my results were nearly to their paper, but were very different from your paper results.(For example, I used SSKD and got acc 72.46, SSKD paper 72.57, your paper 78.21). Could you tell me why? Thanks!

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.