William Falcon 67dc9bc135 fix 16 bit for TPU (#1020) ...

* tpu 16 bit

* tpu 16 bit

* tpu 16 bit

2020-03-03 00:26:59 -05:00

..

__init__.py

update Docs [links & formatting] (#769)

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

auto_mix_precision.py

Trainer cleanup (#934)

2020-02-27 16:21:14 -05:00

callback_config.py

fixes test issues on ddp (#1017)

2020-03-02 21:50:38 -05:00

callback_hook.py

consolidate callbacks and hooks (#950)

2020-03-02 23:51:32 -05:00

data_loading.py

Trainer cleanup (#934)

2020-02-27 16:21:14 -05:00

distrib_data_parallel.py

handle keyboard interrupt for ddp .test() (#1019)

2020-03-02 23:38:47 -05:00

distrib_parts.py

handle keyboard interrupt for ddp .test() (#1019)

2020-03-02 23:38:47 -05:00

evaluation_loop.py

consolidate callbacks and hooks (#950)

2020-03-02 23:51:32 -05:00

ignored_warnings.py

Dim 0 warning (#256)

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

logging.py

Trainer cleanup (#934)

2020-02-27 16:21:14 -05:00

model_hooks.py

Trainer cleanup (#934)

2020-02-27 16:21:14 -05:00

trainer.py

fix 16 bit for TPU (#1020)

2020-03-03 00:26:59 -05:00

training_io.py

handle keyboard interrupt for ddp .test() (#1019)

2020-03-02 23:38:47 -05:00

training_loop.py

consolidate callbacks and hooks (#950)

2020-03-02 23:51:32 -05:00

training_tricks.py

Trainer cleanup (#934)

2020-02-27 16:21:14 -05:00