svdesai / few_bit

Few Bit Supervision

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

few_bit

Few Bit Supervision

Download cifar100.zip and put it in a new directory: 'data/'

CIFAR100 Download link: https://drive.google.com/file/d/12lUydpkvKBLCoNBvDNsZAtrkrRyW8QuH/view?usp=sharing

Tasks Done:

  • One bit supervision code added
  • Our Custom Loss Function added
  • CIFAR100 support

Tasks Pending:

  • Supporting other datasets
  • Adding few bit supervision
  • Margin based hard negative mining
  • Label Smoothing

How to run:

python3 train_stage_0.py
python3 train_stage_n.py

About

Few Bit Supervision


Languages

Language:Python 100.0%