#! /usr/bin/python # -*- coding: utf-8 -*- from __future__ import absolute_import, division, print_function __all__ = ['Adadelta', 'Adagrad', 'Adam', 'Admax', 'Ftrl', 'Nadam', 'RMSprop', 'SGD', 'Momentum', 'Lamb', 'LARS'] # Add module aliases # learning_rate=0.001, rho=0.95, epsilon=1e-07, name='Adadelta' Adadelta = None # learning_rate=0.001, initial_accumulator_value=0.1, epsilon=1e-07,name='Adagrad' Adagrad = None # learning_rate=0.001, beta_1=0.9, beta_2=0.999, epsilon=1e-07, amsgrad=False,name='Adam' Adam = None # learning_rate=0.001, beta_1=0.9, beta_2=0.999, epsilon=1e-07, name='Adamax' Admax = None # learning_rate=0.001, learning_rate_power=-0.5, initial_accumulator_value=0.1, # l1_regularization_strength=0.0, l2_regularization_strength=0.0, name='Ftrl',l2_shrinkage_regularization_strength=0.0 Ftrl = None # learning_rate=0.001, beta_1=0.9, beta_2=0.999, epsilon=1e-07, name='Nadam', Nadam = None # learning_rate=0.001, rho=0.9, momentum=0.0, epsilon=1e-07, centered=False,name='RMSprop' RMSprop = None # learning_rate=0.01, momentum=0.0, nesterov=False, name='SGD' SGD = None # learning_rate, momentum, use_locking=False, name='Momentum', use_nesterov=False Momentum = None def Lamb(**kwargs): raise Exception('Lamb optimizer function not implemented') def LARS(**kwargs): raise Exception('LARS optimizer function not implemented')