pytorch-lightning/pytorch_lightning/callbacks/__init__.py at clean_docs - pytorch-lightning - Gitea: Git with a cup of tea

wassname/pytorch-lightning

mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-08-26 11:24:41 +08:00

Files

T

Stanislav 73cf47112e Gradient accumulation callback (#150 )

* Gradient accumulation callback

* little test case

* typo

* import fix

* method name fix

* fix epochs indexing from 1

* better code style

* code style fix v2 :/

* change interface

* fix Trainre new api in tests

* trainer api bug fix

* new raising error, new update method

* extentions tests

* a little better tests

* typo fix

* flack8 better

* using scheduler for int and dict

* typo

* firs epoch bug fix

* test update

* empty dict exception

* floats check

* codestyle fix

* grad counting test

* someday, i will install normal linter

* add more checks

* Update test_models.py

* Update test_models.py

* Update test_models.py

* Update test_models.py

* Update test_models.py

* Update test_models.py

* Update test_models.py

2019-08-30 10:56:14 -04:00

8 lines

184 B

Python

Raw Permalink Blame History

 from .pt_callbacks import EarlyStopping, ModelCheckpoint, GradientAccumulationScheduler
 __all__ = [
     'EarlyStopping',
     'ModelCheckpoint',
     'GradientAccumulationScheduler',
 ]