open-mmlab / mmaction2

OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark
https://mmaction2.readthedocs.io
Apache License 2.0
4.21k stars 1.23k forks source link

Using classification head inside backbone [Feature] #2728

Open HermannStettin opened 11 months ago

HermannStettin commented 11 months ago

What is the problem this feature will solve?

I want to implement an early exit strategy using MViTv2 for my task. At the beginning my Idea is to simply attach a standard MViTHead after some MultiScaleBlock's.

What is the feature?

I looked up in documentation and code, but didn't find any hints. My question is how can I use classification head inside a backbone?

What alternatives have you considered?

No response