Rohit Gupta 7ca41734da Add `dataloader_idx` to batch transfer hooks (#6241) ... * replace with kwargs * chlog * fix * add test * fix * device * deepspeed * pep * optional * docs * bc * comments * pep * mypy * pep * Apply suggestions from code review * kwargs * docs * . * . * 1.3 -> 1.4 * kwargs -> step_kwargs		2021-05-13 23:03:55 +05:30
..
conf	…
data	`TrainerState` refactor [5/5] (#7173)	2021-05-04 12:50:56 +02:00
__init__.py	…
test_amp.py	`TrainerState` refactor [5/5] (#7173)	2021-05-04 12:50:56 +02:00
test_cpu.py	`TrainerState` refactor [5/5] (#7173)	2021-05-04 12:50:56 +02:00
test_gpu.py	make gpus=str in Trainer consistent with command line parsing of string (#6388)	2021-05-04 09:56:27 +00:00
test_grad_norm.py	`TrainerState` refactor [5/5] (#7173)	2021-05-04 12:50:56 +02:00
test_hooks.py	Add `dataloader_idx` to batch transfer hooks (#6241)	2021-05-13 23:03:55 +05:30
test_horovod.py	`TrainerState` refactor [5/5] (#7173)	2021-05-04 12:50:56 +02:00
test_hparams.py	fix save_hyperparameters(container) if container is empty (#7268)	2021-04-30 13:38:42 +01:00
test_onnx.py	…
test_restore.py	`TrainerState` refactor [5/5] (#7173)	2021-05-04 12:50:56 +02:00
test_sync_batchnorm.py	`TrainerState` refactor [5/5] (#7173)	2021-05-04 12:50:56 +02:00
test_torchscript.py	…
test_tpu.py	Use `torch.nn.utils.clip_grad_norm_` and add `clip_grad_by_value` support for TPU (#7025)	2021-05-07 16:41:39 +00:00