Skip to content
This repository was archived by the owner on Mar 23, 2023. It is now read-only.

grad is none when run gpt2 with pipeline parallelism only #204

@lin88lin8850

Description

@lin88lin8850

🐛 Describe the bug

image

Environment

torch==1.12.0a0+8a1a93a
num_gpu=4
pipeline=4
model = gpt2-small

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions