-
Notifications
You must be signed in to change notification settings - Fork 597
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
[Feature] Support STARK training #434
Conversation
|
Codecov Report
@@ Coverage Diff @@
## master #434 +/- ##
==========================================
+ Coverage 71.23% 72.01% +0.78%
==========================================
Files 119 121 +2
Lines 6748 6954 +206
Branches 1294 1326 +32
==========================================
+ Hits 4807 5008 +201
Misses 1546 1546
- Partials 395 400 +5
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
This reverts commit 6dd98ad.
The refactored code of STARK testing is ready, I will create PR of that after this PR is merged. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
In the forward_train of stark, we only need to
- extract the features of imgs,
- pass the feature into the head
- get the training targets from head
- compute loss
you can refer to siamrpn
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
If there is anything that makes you confused, you can discuss it with me.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Generally speaking, the model
can only call the general API of head
, such as head.init_weights
, head.loss
and so on.
The model
should not call the unique attribute of head
, such as head.run_cls_head
, head.run_reg_head
, head.cls_head
, head.reg_head
and so on.
…g into stark_train_new
No description provided.