Matt PainterandWilliam Falcon 6b667b1237 Fix/test pass overrides (#918) ...

* Fix test requiring both test_step and test_end

* Add test

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

2020-02-24 22:33:11 -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

Clean up dataloader logic (#926)

2020-02-24 22:23:25 -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

Fix/test pass overrides (#918)

2020-02-24 22:33:11 -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

Clean up dataloader logic (#926)

2020-02-24 22:23:25 -05:00

trainer.py

Clean up dataloader logic (#926)

2020-02-24 22:23:25 -05:00

training_io.py

Add tags to the rendezvous calls for TPU. (#921)

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

training_loop.py

Clean up dataloader logic (#926)

2020-02-24 22:23:25 -05:00

training_tricks.py

Refactor callbacks (#776)

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