Message Boards Message Boards

Use custom loss function in NetTrain?

POSTED BY: Iman Nazari
2 Replies

For 11.0 and 11.1 there is only one way of defining a custom loss: build it out of existing net layers (any net that has a scalar output can be used for training). There are many more layers coming in 11.1 which will allow for a great deal more flexibility in defining your own loss functions (including the loss you are trying to define above).

We are looking into supporting users writing loss functions that cannot be written as compositions of existing layers, but this will only be ready earliest 11.2.

Could you demonstrate how to construct a "mean squared relative error" loss function using existing net layers in 11.0 (if possible) ?

POSTED BY: Daniel George
Reply to this discussion
Community posts can be styled and formatted using the Markdown syntax.
Reply Preview
Attachments
Remove
or Discard

Group Abstract Group Abstract