tests/test_optimizer - hpcaitech/ColossalAI - Morph

SIGN IN SIGN UP

hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

41377 0 1 Python

	_utils.py	13.1 KB
	test_adam_kernel.py	5.9 KB
	test_adam_optim.py	2.6 KB
	test_dist_adafactor.py	16.0 KB
	test_dist_came.py	12.2 KB
	test_dist_galore.py	10.0 KB
	test_dist_lamb.py	9.4 KB
	test_lr_scheduler.py	661 B
	test_nvme.py	1.9 KB