    def test_remove_bridge_module_state_dic(self):
        '''
        for remove BridgeFormer module parameters
        '''
        x = {'text_model.embeddings.word_embeddings.weight': torch.randn(2, 4),
             'text_model.embeddings.position_embeddings.weight': torch.randn(2, 4)}
        y = {'text_model.embeddings.word_embeddings.weight': torch.randn(2, 4),
             'text_model.embeddings.position_embeddings.weight': torch.randn(2, 4),
             'bridge.text_model.embeddings.position_embeddings.weight': torch.randn(2, 4),
             'video_model.blocks.0.crossattn.proj.bias': torch.randn(2, 4)}
        out = remove_bridge_module_state_dic(y, x)
        self.assertEqual(x.keys(), out.keys())
for remove BridgeFormer module parameters
tests/unittests/models/frozen_in_time/test_frozen_utils.py