Peter IzsakandWilliam Falcon 054a35312d Added max number of steps in Trainer (#728) ...

* Added max number of steps in Trainer

* Added docstring

* Fix flake8 errors

* Clarified docstrings

* Fixed flake8 error

* Added min_steps to Trainer

* Added steps and epochs test

* flake8

* minor fix

* fix steps test in test_trainer

* Split steps test into 2 tests

* Refactor steps test

* Update test_trainer.py

* Minor in test_trainer.py

* Update test_trainer.py

* Address PR comments

* Minor

Co-authored-by: William Falcon <waf2107@columbia.edu>

2020-02-18 11:23:22 -05:00

..

__init__.py

update Docs [links & formatting] (#769)

2020-02-09 17:39:10 -05:00

auto_mix_precision.py

Enable TPU support (#868)

2020-02-17 16:01:20 -05:00

callback_config.py

Refactor callbacks (#776)

2020-02-16 00:03:05 -05:00

data_loading.py

Enable TPU support (#868)

2020-02-17 16:01:20 -05:00

distrib_data_parallel.py

Enable TPU support (#868)

2020-02-17 16:01:20 -05:00

distrib_parts.py

fix tpu transfer bug 2

2020-02-17 17:47:16 -05:00

evaluation_loop.py

Enable TPU support (#868)

2020-02-17 16:01:20 -05:00

ignored_warnings.py

Dim 0 warning (#256)

2019-09-26 13:20:54 -04:00

logging.py

Allow user to specify 'step' key while logging metrics (#808)

2020-02-15 23:35:23 -05:00

model_hooks.py

rename trainer modules, drop _mixin (#571)

2019-12-04 11:39:14 -05:00

trainer.py

Added max number of steps in Trainer (#728)

2020-02-18 11:23:22 -05:00

training_io.py

rename logging -> loggers (#767)

2020-02-01 15:47:58 -05:00

training_loop.py

Added max number of steps in Trainer (#728)

2020-02-18 11:23:22 -05:00

training_tricks.py

Refactor callbacks (#776)

2020-02-16 00:03:05 -05:00