How to save features extracted by the backbone #428
-
Beta Was this translation helpful? Give feedback.
Answered by
ashwinvaidya17
Jul 4, 2022
Replies: 2 comments
-
You should call the print statement after |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
djdameln
-
Converting to Q&A in Discussions, feel free to continue from there. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
You should call the print statement after
_ = self.backbone(input_tensor)
line (95). It is because hooks are added to the backbone and only when the forward function of the backbone is called, the features dictionary is populated.