• Stars
    star
    159
  • Rank 235,916 (Top 5 %)
  • Language
    Python
  • License
    MIT License
  • Created over 5 years ago
  • Updated about 4 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

An unofficial implementation of 《Deep Mutual Learning》 by Pytorch to do classification on cifar100.

Deep-Mutual-Learning

This is an unofficial implementation of Deep Mutual Learning by Pytorch to do classification on cifar100.
The algorithm was proposed in 《Deep Mutual Learning》 (CVPR 2017).

Dependence

Pytorch 1.0.0
tensorboard 1.14.0

Overview

Overview of the algorithm:

Usage

The default network for DML is ResNet32.
Train 2 models using DML by main.py:

python train.py --model_num 2

Use tensorboard to monitor training process on choosing port:

tensorboard --logdir logs --port 6006

Result

Network ind_avg_acc Dml_avg_acc
ResNet32 69.83% 71.03%