/gradient_adjustment

Primary LanguagePython

Learning to Minimize Remainder in Supervised Learning

Code for project: "Learning to Minimize Remainder in Supervised Learning".

Requirements and Environment

  • Main dependency: PyTorch 1.4.0+
  • The code is tested under Ubuntu 1804 LTS with Python 3.7.

Tasks for Evaluation

3D Example

The code and script that visualize the convergence on a 3D example problem are in folder 3d_example.

CIFAR

The code and script that run experiments on CIFAR-10 and CIFAR-100 are in folder cifar.

ImageNet

The code and script that run experiments on ImageNet are in folder imagenet.