Skip to content

Can we support class weight in the CEWithChunkedOutputLoss class #1746

Open
@ye-jin-shop

Description

@ye-jin-shop

I am trying to add weights to the loss function. I think it would be nice to have it in the class function? The original class CrossEntropyLoss has weight as an arg.

Metadata

Metadata

Assignees

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