attacks/privacy_attacks.py [263:272]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    if params.dataset=='imagenet':
        new_state_dict = OrderedDict()
        for k, v in state_dict_final["model"].items():
            if k[:7]=='module.': # remove `module.`
                new_state_dict[k[7:]] = v
            else:
                new_state_dict[k]=v
        final_model.load_state_dict(new_state_dict)
    else:
        final_model.load_state_dict(state_dict_final['model'])
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



attacks/privacy_attacks.py [292:301]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        if params.dataset=='imagenet':
            new_state_dict = OrderedDict()
            for k, v in state_dict_final["model"].items():
                if k[:7]=='module.': # remove "module.""
                    new_state_dict[k[7:]] = v
                else:
                    new_state_dict[k]=v
            final_model.load_state_dict(new_state_dict)
        else:
            final_model.load_state_dict(state_dict_final['model'])
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -