868b172f05
* Make training_epoch_end behave like validation_epoch_end + minor fixes in docstrings. * Minor fixes (Borda's comments). * Detach tensors in batch_output (to avoid possible memory leak) + doc fix. Co-authored-by: Jean-Baptiste SCHIRATTI <jean-baptisteschiratti@MacBook-Pro-de-Jean-Baptiste.local> |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| auto_mix_precision.py | ||
| callback_config.py | ||
| callback_hook.py | ||
| data_loading.py | ||
| deprecated_api.py | ||
| distrib_data_parallel.py | ||
| distrib_parts.py | ||
| evaluation_loop.py | ||
| ignored_warnings.py | ||
| logging.py | ||
| model_hooks.py | ||
| optimizers.py | ||
| supporters.py | ||
| trainer.py | ||
| training_io.py | ||
| training_loop.py | ||
| training_tricks.py | ||