Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

could I set a loss for neck? #792

Closed
YilanWang opened this issue Apr 21, 2022 · 3 comments
Closed

could I set a loss for neck? #792

YilanWang opened this issue Apr 21, 2022 · 3 comments
Labels
help wanted Extra attention is needed

Comments

@YilanWang
Copy link

I want to set a special auxiliary loss for neck features, however I don't read all mmcls repo. By the experiences in using mmdet, it seemed like I need to set many heads for rpn(rpn_head) and rcnn(roi_head).
Could you tell me, is it useful for setting a loss for neck? Many thanks~~~

@YilanWang YilanWang added the help wanted Extra attention is needed label Apr 21, 2022
@Ezra-Yu
Copy link
Collaborator

Ezra-Yu commented Apr 21, 2022

Yes, your intuition is right, just modify the loss in the model.head such as neck in resnet

@YilanWang
Copy link
Author

Yes, your intuition is right, just modify the loss in the model.head such as neck in resnet

many thanks for reply, so could I set 2 heads for a net? If so, should I implement it like
head=dict( type='LinearClsHead', xxx), dict( type='yyy') )
or other foramt?
If not, could you tell me how to implement a auxiliary loss?
Many many thanks~~

@Ezra-Yu
Copy link
Collaborator

Ezra-Yu commented Apr 21, 2022

The master/dev branch does not yet support this feature, you can refer to this mutil task PR.

@Ezra-Yu Ezra-Yu closed this as completed Apr 22, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
help wanted Extra attention is needed
Projects
None yet
Development

No branches or pull requests

2 participants