Skip to content

[REQUEST] Pipeline Parallelism support multi optimizer to train #6951

Open
@whcjb

Description

Is your feature request related to a problem? Please describe.
i want to train big model gan, must shard to multi-gpu, but the pipeline_module seems not support multi-optimizer
Describe the solution you'd like
support

Describe alternatives you've considered
can control flow to which layer

Additional context

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions