tests/test_optimizer - hpcaitech/ColossalAI - Morph

SIGN IN SIGN UP

hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

41373 0 1 Python

	_utils.py	9.8 KB
	test_adam_kernel.py	5.9 KB
	test_adam_optim.py	3.1 KB
	test_dist_adafactor.py	18.7 KB
	test_dist_came.py	15.7 KB
	test_dist_galore.py	11.2 KB
	test_dist_lamb.py	10.0 KB
	test_lr_scheduler.py	661 B
	test_nvme.py	1.9 KB