SIGN IN SIGN UP
hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 36 Python

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

* add custom policy

* update assert
X
Xuanlei Zhao committed
ac2797996b362e5bded4d0eec18ef96efc12b086
Parent: 451c346
Committed by GitHub <noreply@github.com> on 9/15/2023, 9:53:13 AM