tests/models - Lightning-AI/pytorch-lightning - Morph

SIGN IN SIGN UP

Lightning-AI / pytorch-lightning UNCLAIMED

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

30975 0 0 Python

	conf
	data
	__init__.py
	test_amp.py	7.7 KB
	test_cpu.py	11.7 KB
	test_gpu.py	12.7 KB
	test_grad_norm.py	4.2 KB
	test_hooks.py	24.3 KB
	test_horovod.py	14.1 KB
	test_hparams.py	23.3 KB
	test_onnx.py	4.9 KB
	test_restore.py	20.1 KB
	test_sync_batchnorm.py	3.8 KB
	test_torchscript.py	6.1 KB
	test_tpu.py	14.9 KB