Skip to content
This repository has been archived by the owner on Nov 21, 2023. It is now read-only.

How to change classification loss function #523

Closed
PacteraOliver opened this issue Jun 25, 2018 · 1 comment
Closed

How to change classification loss function #523

PacteraOliver opened this issue Jun 25, 2018 · 1 comment

Comments

@PacteraOliver
Copy link

I found #438 and #257 asked about the loss function issues.

I want to know how to change the loss function for a certain network.

I found that the loss function is now written in caffe2 select_smooth_l1_loss_op.h

Is there a way to change it or define by our self?

Thank you.

@ir413
Copy link
Contributor

ir413 commented Jun 26, 2018

Hi @PacteraOliver, you can change the classification loss function here. You may also find it useful to look at the following functions that are used to define the losses for different tasks: fast_rcnn_losses, mask_rcnn_losses and keypoint_rcnn_losses.

@ir413 ir413 closed this as completed Jun 26, 2018
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants