mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
pytorch-image-models/timm/optim/__init__.py

16 lines
484 B
Python
Raw Normal View History

Add lamb/lars to optim init imports, remove stray comment
2021-08-19 13:55:02 +08:00
from .adabelief import AdaBelief
Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.
2020-10-10 08:24:08 +08:00
from .adafactor import Adafactor
from .adahessian import Adahessian
Add lamb/lars to optim init imports, remove stray comment
2021-08-19 13:55:02 +08:00
from .adamp import AdamP
from .adamw import AdamW
from .lamb import Lamb
from .lars import Lars
Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.
2020-10-10 08:24:08 +08:00
from .lookahead import Lookahead
A few optimizer comments, dead import, missing import
2021-08-18 09:01:33 +08:00
from .madgrad import MADGRAD
Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.
2020-10-10 08:24:08 +08:00
from .nadam import Nadam
Add Nvidia's NovogGrad impl from Jasper (cleaner/faster than current) and Apex Fused optimizers
2019-08-30 06:21:38 +08:00
from .nvnovograd import NvNovoGrad
Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.
2020-10-10 08:24:08 +08:00
from .radam import RAdam
from .rmsprop_tf import RMSpropTF
Add `adamp` and 'sgdp' optimizers. Update requirements.txt Update optim_factory.py Add `adamp` optimizer Update __init__.py copy files of adamp & sgdp Create adamp.py Update __init__.py Create sgdp.py Update optim_factory.py Update optim_factory.py Update requirements.txt Update adamp.py Update sgdp.py Update sgdp.py Update adamp.py
2020-07-22 10:02:05 +08:00
from .sgdp import SGDP
adabelief
2021-04-05 11:48:15 +08:00
from .optim_factory import create_optimizer, create_optimizer_v2, optimizer_kwargs