BaoWangMath / LaplacianSmoothing-GradientDescent

The code for the paper: https://arxiv.org/abs/1806.06317
24 stars 6 forks source link

LaplacianSmoothing-GradientDescent

Code for the paper: https://arxiv.org/abs/1806.06317

Quadratic-Function

This directory contains the LS-GD code for quadratic function

Small-BatchSize

This directory contains the LS-GD code for training LeNet on MNIST data with small batch size

ResNet

This directory contains the LS-GD code for training ResNet

Tensorflow-LSSGD

This directory contains the tensorflow implementation of LS-SGD

Dependence

https://github.com/locuslab/pytorch_fft

PyTorch version <= 0.3.0