Hadrien Mary be244560b2 Callbacks [wip] (#889) ...

* Add callback system + associated test

* Add trainer and pl_module args to callback methods

* typing

* typo in docstring

* Switch to on_.*_start()

* fix on_test_start

* fix the mess after rebasing

2020-02-25 23:17:27 -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

Callbacks [wip] (#889)

2020-02-25 23:17:27 -05:00

callback_hook.py

Callbacks [wip] (#889)

2020-02-25 23:17:27 -05:00

data_loading.py

Fixing tests (#936)

2020-02-25 13:06:24 -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

Callbacks [wip] (#889)

2020-02-25 23:17:27 -05:00

ignored_warnings.py

Dim 0 warning (#256)

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

logging.py

Add support for multiple loggers (#903)

2020-02-25 14:52:39 -05:00

model_hooks.py

Clean up dataloader logic (#926)

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

trainer.py

Callbacks [wip] (#889)

2020-02-25 23:17:27 -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

Callbacks [wip] (#889)

2020-02-25 23:17:27 -05:00

training_tricks.py

Callbacks [wip] (#889)

2020-02-25 23:17:27 -05:00