OpenGVLab / UniFormerV2

[ICCV2023] UniFormerV2: Spatiotemporal Learning by Arming Image ViTs with Video UniFormer
https://arxiv.org/abs/2211.09552
Apache License 2.0
294 stars 19 forks source link

Evaluation using mit - getting the embeddings #27

Closed anjugopinath closed 1 year ago

anjugopinath commented 1 year ago

I am performing evaluation using mit. This is my test.sh file:

image

I want to get the output (embeddings) of the second last layer. I will be using pytorch hooks to do this. Which file will give me the access to the model during evaluation?

Andy1621 commented 1 year ago

The model code is here.