search

Anirudh257 / strm

[CVPR 2022] Official Pytorch Implementation for "Spatio-temporal Relation Modeling for Few-shot Action Recognition". SOTA Results for Few-shot Action Recognition
https://anirudh257.github.io/strm/
99 stars 16 forks source link

Missing key(s) in state_dict #37

Open Angel-la opened 7 months ago

Angel-la commented 7 months ago

Hello,

I encountered an issue while using the provided checkpoints from the linked Google Drive. Several keys related to the Transformer are missing from the state sictionary:

Missing key(s) in state_dict: "transformers.1.pe.pe", "transformers.1.k_linear.weight", "transformers.1.k_linear.bias", "transformers.1.v_linear.weight", "transformers.1.v_linear.bias", "transformers.1.norm_k.weight", "transformers.1.norm_k.bias", "transformers.1.norm_v.weight", "transformers.1.norm_v.bias"

I tried both checkpoints for Kinetics and SSV2.

Could you please provide updated checkpoints?

Anirudh257 commented 1 month ago

Hi, sorry for the late response. I believe that this is due to different environment than https://github.com/Anirudh257/strm/blob/main/environment.yaml

Can you check if this works?