search

zhoumingyi / DaST

A novel data-free model stealing method based on GAN
123 stars 24 forks source link

question in dast_cifar10.py #14

Open jiachengc opened 5 months ago

jiachengc commented 5 months ago

Hi, thanks for this amazing code. I have a question about line 30 in dast_cifar10.py.

what is this "vgg_sun", I cannot really figure what is mean. Is it a typo? https://github.com/zhoumingyi/DaST/issues.

Thanks

jiachengc commented 5 months ago

After I change this line from "from vgg_sun import VGG" to "from vgg import VGG". It gives me error "raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for VGG: Missing key(s) in state_dict: "classifier.1.weight", "classifier.1.bias", "classifier.4.weight", "classifier.4.bias", "classifier.6.weight", "classifier.6.bias". Unexpected key(s) in state_dict: "classifier.weight", "classifier.bias""

zhoumingyi commented 5 months ago

Sorry. It may be because I uploaded the version of testing VGG models from others (some people cannot reproduce the results on their own VGG models). You can try to re-train the VGG model. I will fix it after returning to work (I am on leave for several days).