Function load_decoder_and_message

wm_utils.py:142–149

(msg_len, sdir)

    return (1. - err) * 100

def load_decoder_and_message(msg_len, sdir):
    sdict = torch.load(os.path.join(sdir, f'CLIP-MsgDecoder-{msg_len}.pkl'), map_location='cpu', weights_only=True)
    model = CLIPWatermarker(msg_len=msg_len)
    model.load_state_dict(sdict.pop('model'))
    for k, v in model.named_parameters():
        v.requires_grad = False    
    message = random.choice(sdict['data'])[1]
    return model.cuda(), message.cuda()