rorschachchen / deepsigns-torch

DeepSigns

This repository provides an pytorch API for DeepSign framework. DeepSign is an end-to-end watermarking framework for IP protection of deep neural networks. The paper 'DeepSigns: A Generic Watermarking Framework for IP Protection of Deep Learning Models' is available on arXiv: https://arxiv.org/abs/1804.00750.

Contact Info:

If you have any questions, please email [email protected]

deepsigns-torch's Issues

utils.py line 127: 'bool' object has no attribute 'nonzero'

Hi, I have tried runing python run_whitebox.py, but I got this:

in line 127

dataset.targets == target_class returns False, which does not have member function nonzero().

I have environment with

Python 3.9.0
torchvision 0.12.0
pytorch 1.11.0
pandas 1.4.2
scipy 1.8.0

is Blackbox code correct ?

@RorschachChen Have you tried the blackbox code (ran it and checked if it actually works)? For me marked and unmarked model the output is same (both are true for the verification)