core.trainers.mixins.distributed_parallel
core.trainers.mixins.distributed_parallel
Mixin for correctly saving fsdp
Classes
| Name | Description |
|---|---|
| DistributedParallelMixin | Mixin for correctly saving fsdp |
DistributedParallelMixin
core.trainers.mixins.distributed_parallel.DistributedParallelMixin()Mixin for correctly saving fsdp