"vscode:/vscode.git/clone" did not exist on "33d7e89c42e0fe7b4a277d7a5bae12ba14828dd8"
__init__.py 521 Bytes
Newer Older
Tim Dettmers's avatar
Tim Dettmers committed
1
2
3
4
5
6
7
8
9
# Copyright (c) Facebook, Inc. and its affiliates. 
#   
# This source code is licensed under the MIT license found in the 
# LICENSE file in the root directory of this source tree.
from .adam import Adam, Adam8bit, Adam32bit
from .sgd import SGD, SGD8bit, SGD32bit
from .lars import LARS, LARS8bit, LARS32bit, PytorchLARS
from .lamb import LAMB, LAMB8bit, LAMB32bit
from .rmsprop import RMSprop, RMSprop8bit, RMSprop32bit
10
from .adagrad import Adagrad, Adagrad8bit, Adagrad32bit
Tim Dettmers's avatar
Tim Dettmers committed
11
from .optimizer import GlobalOptimManager