Skip to content

Allow loss functions to receive additional parameters #28

@ChenglongWang

Description

@ChenglongWang

Currently, loss function is only allowed to receive network's output and ground-truth.
To increase the flexibility of loss function, additional parameters should also be visible to loss function.

Solution:
Add additional key like additional_loss_data to pass the additional data to loss function.

Metadata

Metadata

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