Leonard Lausen 597b309f2e Fix `Trainer.plugins` type declaration (#7288) ... * Fix trainer.plugins type declaration * Don't ClusterEnvironment(Plugin) * fix import error, yapf formatter * Add test Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>		2021-05-04 08:42:57 +02:00
..
__init__.py	New metric classes (#1326) (#1877)	2020-05-19 11:05:07 -04:00
distributed.py	Fixed PYTHONPATH for ddp test model (#4528)	2020-12-05 20:09:47 +00:00
test_all_gather_grad.py	Automatically find and run special tests (#6669)	2021-03-26 17:04:59 +00:00
test_apply_func.py	formatting tests1/n (#5843)	2021-02-06 08:22:10 -05:00
test_apply_func_torchtext.py	fix importing torchtext batch (#6365)	2021-03-05 21:39:52 +01:00
test_argparse.py	Fix argparse docs (#7148)	2021-04-22 02:13:00 +02:00
test_cli.py	Fix `Trainer.plugins` type declaration (#7288)	2021-05-04 08:42:57 +02:00
test_distributed.py	[fix] Better support for rank_zero_only setting for SLURM and torchelastic (#6802)	2021-04-07 12:25:13 +01:00
test_dtype_device_mixin.py	[feat] Add better support for predict + ddp 2/3 (#7215)	2021-04-27 08:46:45 -04:00
test_imports.py	fix import and typo in AMP (#4871)	2020-11-26 23:45:52 +01:00
test_memory.py	Make move_metrics_to_cpu work recursively (#6007)	2021-02-16 21:52:42 +00:00
test_parsing.py	add forgotten test in #7240 (#7283)	2021-05-03 23:56:30 +00:00
test_seed.py	Fixed `num_sanity_val_steps` affecting reproducibility of training data shuffling (#7014)	2021-04-27 09:51:39 +00:00
test_upgrade_checkpoint.py	formatting tests1/n (#5843)	2021-02-06 08:22:10 -05:00
test_xla_device_utils.py	[TPU] update is_tpu_exists utils internal logic to rely on xmp.spawn (#6719)	2021-03-29 18:59:20 +01:00