Function remove_prefix_module

TinyCLIP/src/training/main.py:408–410

(state_dict)

    else:

        def remove_prefix_module(state_dict):
            # remove the first or the second module
            return convert_to_new_checkpoint(state_dict)

        def add_prefix_module(state_dict):
            if all(map(lambda x: not x.startswith('module.'), state_dict.keys())):