[AutoBackbone] Improve API#20407
Conversation
|
The documentation is not available anymore as the PR was closed or merged. |
c7841cf to
55e13ea
Compare
|
Pinging @ydshieh on this PR as I made 2 updates to test_modeling_common.py to support models which output a tuple of tensors as their first output. I've updated |
|
@michaelbenayoun as seen on the CI, the torch fx tests fail because backbones aren't supported yet. Could you add support for them in a separate PR? |
Looks good for me regarding the tests. I would personally put the check |
|
@ydshieh for some reason the CI isn't run when I push new commits, do you know why? |
|
Not really, try again with a new empty commit? git commit --allow-empty -m "empty commit to trigger CI" |
* Add hidden states and attentions to backbone outputs * Update ResNet * Fix more tests * Debug test * Fix test_determinism * Fix test_save_load * Remove file * Disable fx tests * Test * Add fx support for backbones Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
What does this PR do?
As backbones themselves have hidden states and optional attentions, this PR adds them to the
BackboneOutput.This way, frameworks that leverage backbones can return hidden states/attentions of the backbone if the user specifies
output_hidden_states=Trueoroutput_attentions=True.To do: