| .. |
|
__init__.py
|
Allow user to select individual TPU core to train on (#1729)
|
2020-05-17 16:30:54 -04:00 |
|
auto_mix_precision.py
|
support for native amp (#1561)
|
2020-04-23 14:47:08 -04:00 |
|
callback_config.py
|
protect progress bar callback (#1855)
|
2020-05-25 07:49:23 -04:00 |
|
callback_hook.py
|
remove extra kwargs from Trainer init (#1820)
|
2020-05-17 09:14:54 -04:00 |
|
data_loading.py
|
add warning for shuffling in test/val (#1865)
|
2020-05-18 09:53:02 -04:00 |
|
deprecated_api.py
|
protect progress bar callback (#1855)
|
2020-05-25 07:49:23 -04:00 |
|
distrib_data_parallel.py
|
Fix user warning produced by apex + scheduler combination (#1873)
|
2020-05-22 07:19:37 -04:00 |
|
distrib_parts.py
|
Remove unused param tpu_core_idx (#1948)
|
2020-05-25 16:04:53 -04:00 |
|
evaluation_loop.py
|
Allow user to select individual TPU core to train on (#1729)
|
2020-05-17 16:30:54 -04:00 |
|
ignored_warnings.py
|
Dim 0 warning (#256)
|
2019-09-26 13:20:54 -04:00 |
|
logging.py
|
[WIP] Reduction when batch size < num gpus (#1609)
|
2020-05-02 11:01:44 -04:00 |
|
lr_finder.py
|
protect progress bar callback (#1855)
|
2020-05-25 07:49:23 -04:00 |
|
model_hooks.py
|
Fix typo (#1750)
|
2020-05-07 09:25:54 -04:00 |
|
optimizers.py
|
Fix user warning produced by apex + scheduler combination (#1873)
|
2020-05-22 07:19:37 -04:00 |
|
seed.py
|
Option to provide seed to random generators to ensure reproducibility (#1572)
|
2020-05-12 07:53:20 -04:00 |
|
supporters.py
|
Fixed configure optimizer from dict without "scheduler" key (#1443)
|
2020-04-10 11:43:06 -04:00 |
|
trainer.py
|
Gen ddp support (#1961)
|
2020-05-26 19:02:30 -04:00 |
|
training_io.py
|
replace Hparams by init args (#1896)
|
2020-05-24 18:59:08 -04:00 |
|
training_loop.py
|
early stopping checks on_validation_end (#1458)
|
2020-05-25 17:33:00 +00:00 |
|
training_tricks.py
|
replace Hparams by init args (#1896)
|
2020-05-24 18:59:08 -04:00 |
460ab5485e