Adrian Wälchli 2f3491a739 Fix saving FSDP checkpoint when world size = 1 and torch <= 2.0 (#18371)		2023-08-23 06:45:20 -04:00
..
accelerators	docformatter: config with black (#18064)	2023-08-09 10:44:20 -04:00
graveyard	[TPU] Proper half-precision implementation for XLA (#18213)	2023-08-11 11:37:41 -04:00
helpers	[pre-commit.ci] pre-commit suggestions (#17983)	2023-08-08 16:26:06 +02:00
loggers	[pre-commit.ci] pre-commit suggestions (#17983)	2023-08-08 16:26:06 +02:00
plugins	Accommodate FSDP full-precision `param_dtype` training with PyTorch < 2.0 (#18278)	2023-08-14 12:22:26 +02:00
strategies	Fix saving FSDP checkpoint when world size = 1 and torch <= 2.0 (#18371)	2023-08-23 06:45:20 -04:00
utilities	improve debug failing warning test (#18271)	2023-08-12 07:59:52 -04:00
__init__.py	…
conftest.py	docformatter: config with black (#18064)	2023-08-09 10:44:20 -04:00
run_standalone_tests.sh	…
run_tpu_tests.sh	Fix [TPU] tests (#18140)	2023-07-24 15:13:36 +02:00
test_cli.py	docformatter: config with black (#18064)	2023-08-09 10:44:20 -04:00
test_connector.py	[TPU] Proper half-precision implementation for XLA (#18213)	2023-08-11 11:37:41 -04:00
test_fabric.py	Mention specific param names and devices in warning (#18273)	2023-08-16 00:51:28 +02:00
test_wrappers.py	docformatter: config with black (#18064)	2023-08-09 10:44:20 -04:00