Missing key(s) in state_dict

[Angel-la]

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]

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?