-
Notifications
You must be signed in to change notification settings - Fork 25.2k
Open
Labels
module: cpuCPU specific problem (e.g., perf, algorithm)CPU specific problem (e.g., perf, algorithm)module: cudaRelated to torch.cuda, and CUDA support in generalRelated to torch.cuda, and CUDA support in generalmodule: lossProblem is related to loss functionProblem is related to loss functiontriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
In BCEWithLogitsLoss, weights are unnormalized and used, while in ClassNLLLoss weights are normalized and used.
Some discussion around this can be found here. Is this discrepancy expected? Else if this is an issue I can send a PR to resolve this.
Metadata
Metadata
Assignees
Labels
module: cpuCPU specific problem (e.g., perf, algorithm)CPU specific problem (e.g., perf, algorithm)module: cudaRelated to torch.cuda, and CUDA support in generalRelated to torch.cuda, and CUDA support in generalmodule: lossProblem is related to loss functionProblem is related to loss functiontriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module