Adrian Wälchli 89a5d8fee9 fix auto scale batch size not working with precision=16 (#3045) ... * add test * test * test * add fix * changelog * check batch size changed		2020-08-19 20:41:33 +00:00
..
__init__.py	tasks docs	2020-08-15 22:36:53 -04:00
auto_mix_precision.py	add apex test (#2921)	2020-08-13 10:03:13 -04:00
callback_config.py	clean imports (#2867)	2020-08-08 00:33:51 +02:00
callback_hook.py	updated hooks (#2850)	2020-08-07 09:29:57 -04:00
configuration_validator.py	Fix enforce_datamodule_dataloader_override() for iterable datasets (#2957)	2020-08-13 23:06:17 +02:00
data_loading.py	allow using apex with any PT version (#2865)	2020-08-08 11:07:32 +02:00
deprecated_api.py	…
distrib_data_parallel.py	ddp fix for trainer.test() + add basic ddp tests (#2997)	2020-08-16 11:19:57 -04:00
distrib_parts.py	add apex test (#2921)	2020-08-13 10:03:13 -04:00
evaluation_loop.py	fix result obj dp auto reduce (#3013)	2020-08-17 10:29:39 -04:00
ignored_warnings.py	Support returning python scalars in DP (#1935)	2020-08-07 09:18:29 +02:00
logging.py	Fixes #2972 #2946 (#2986)	2020-08-15 08:36:00 -04:00
lr_finder.py	clean imports (#2867)	2020-08-08 00:33:51 +02:00
model_hooks.py	…
optimizers.py	…
states.py	Add tracking of basic states in Trainer [wip - to-be-merged after v0.9] (#2541)	2020-08-09 06:24:09 -04:00
supporters.py	Save test predictions on multiple GPUs (#2926)	2020-08-14 17:52:43 -04:00
trainer.py	ddp fix for trainer.test() + add basic ddp tests (#2997)	2020-08-16 11:19:57 -04:00
training_io.py	fix auto scale batch size not working with precision=16 (#3045)	2020-08-19 20:41:33 +00:00
training_loop.py	fix result obj dp auto reduce (#3013)	2020-08-17 10:29:39 -04:00
training_tricks.py	add apex test (#2921)	2020-08-13 10:03:13 -04:00