8000 GitHub - congjianluo/TensorflowASGD: The tensorflow ASGD optimizer implement.

More Web Proxy on the site http://driver.im/

Name		Name	Last commit message	Last commit date
Latest commit History 14 Commits
LeNet		LeNet
.gitignore		.gitignore
Readme.md		Readme.md
TensorFlowASGD.py		TensorFlowASGD.py
choose_best_gpu.py		choose_best_gpu.py
pytorch_demo.py		pytorch_demo.py
tensorflow_demo.py		tensorflow_demo.py

Repository files navigation

Tensorflow ASGD Optimizer

Custom Asynchronous Stochastic Gradient Descent Optimizer in Tensorflow.

It works like PyTorch ASGD. https://pytorch.org/docs/stable/_modules/torch/optim/asgd.html#ASGD

You can import it:

from TensorFlowASGD import AsynchronousStochasticGradientDescent

and use replace the optimizer tha you used:

optimizer = AsynchronousStochasticGradientDescent()

A simple LeNet example:

About

The tensorflow ASGD optimizer implement.

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%

0