ColossalAI-Pytorch-lightning icon indicating copy to clipboard operation
ColossalAI-Pytorch-lightning copied to clipboard

Published 20 hours ago • verified publisher icon hpcaitech
→

Metadata

ColossalAI Pytorch-lightning strategy

Usage

from pytorch_lightning.strategies.colossalai import ColossalAIStrategy

trainer = Trainer(..., strategy=ColossalAIStrategy())

← Metadata

23
Stars
6
Forks
Watchers

Owner

verified publisher icon hpcaitech

Metadata

Back