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

[Known Issue] Deepspeed multiple loss support #4

Description

@ncilfone

If running with multiple losses Deepseep will currently fail as retain_graph is not passed to the backward call within the Deepspeed engine thus preventing multiple backward calls within stoke.

Please use a single loss function for the time being until this can be patched -- in most simple situations with multiple losses simply add them...

PR is currently opened to fix this: deepspeedai/DeepSpeed#1149

Metadata

Metadata

Assignees

Labels

bugSomething isn't working

Type

No type

Fields

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