tests/test_shardformer - hpcaitech/ColossalAI

Making large AI models cheaper, faster and more accessible

0 0 0 Python

	test_hybrid_parallel_grad_clip_norm
	test_layer
	test_model
	__init__.py
	test_flash_attention.py	5.6 KB
	test_shard_utils.py	833 B
	test_with_torch_ddp.py	2.6 KB