YellowFin_Pytorch

Optimization algorithm

An optimizer that automatically tunes momentum and learning rate using local quadratic approximation.

auto-tuning momentum SGD optimizer

GitHub

287 stars
13 watching
32 forks
Language: Python
last commit: over 5 years ago

Related projects:

Repository Description Stars
guopengf/auto-fedrl A reinforcement learning-based framework for optimizing hyperparameters in distributed machine learning environments. 15
ikostrikov/pytorch-trpo A PyTorch implementation of an optimization algorithm for continuous control and reinforcement learning tasks 435
jmrichardson/tuneta Automates optimization of technical indicators for machine learning models in finance 421
metaopt/torchopt An efficient library for differentiable optimization built on top of PyTorch. 554
locuslab/optnet A PyTorch module that adds differentiable optimization as a layer to neural networks 517
moskomule/eve.pytorch An implementation of an optimization algorithm inspired by a 2016 research paper 33
xidongwu/federated-minimax-and-conditional-stochastic-optimization This project presents optimization techniques for federated learning and minimax games in the context of machine learning 1
locuslab/qpth A solver for optimizing quadratic programs in deep learning and other fields. 691
yaodongyu/tct An approach to train and optimize machine learning models in a decentralized setting by convexifying the optimization process 4
hui-po-wang/progfed An approach to efficient federated learning by progressively training models on client devices with reduced communication and computation requirements. 20
hyperopt/hyperopt-sklearn Automates search for optimal parameters in machine learning algorithms. 1,594
rahulkidambi/accsgd An implementation of the Accelerated SGD algorithm in PyTorch 215
jameschch/leanparameteroptimization A toolset for parameter optimization in trading algorithms using Lean 57
liyanghart/hyperparameter-optimization-of-machine-learning-algorithms Provides tools and techniques for tuning hyperparameters in machine learning models to improve performance. 1,283
ikostrikov/pytorch-meta-optimizer A PyTorch implementation of meta-learning using gradient descent to adapt to new tasks. 312