update api for better usability#58

FrankLeeeee merged 2 commits intohpcaitech:develop/experimentshpcaitech/ColossalAI:develop/experimentsfrom

FrankLeeeee:develop/experimentsFrankLeeeee/ColossalAI:develop/experimentsCopy head branch name to clipboard