Ir1dXDandWilliam Falcon be83e7515b feat(trainer): add enable_benchmarking option (#803) ...

* feat(trainer): add enable_benchmarking option

closes #370

* Update trainer.py

* Update trainer.py

* Update trainer.py

* Update trainer.py

* Update trainer.py

* Update trainer.py

* Update trainer.py

* Update trainer.py

* add test

* try to make the lint work

* fix typo

* add test, verify torch.backends.cudnn.benchmark

* make lint happy

* make lint happy

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

2020-02-25 15:05:41 -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

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

Fixing tests (#936)

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

feat(trainer): add enable_benchmarking option (#803)

2020-02-25 15:05:41 -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

Fixing tests (#936)

2020-02-25 13:06:24 -05:00

training_tricks.py

Refactor callbacks (#776)

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