[shardformer] add custom policy in hybrid parallel plugin#4718

oahzxl merged 2 commits into

hpcaitech:mainhpcaitech/ColossalAI:mainfrom

oahzxl:custom_policyoahzxl/ColossalAI:custom_policyCopy head branch name to clipboard