def append_module_suffix(state_dict, suffix):
    """
    Append suffixes in a state_dict
    Needed when loading DDP or classy vision models
    """
    state_dict = {f"{suffix}{key}": val for (key, val) in state_dict.items()}
    return state_dict