search

nv-tlabs / datasetGAN_release

MIT License
340 stars 44 forks source link

checkpoint #8

Open ZhouCX117 opened 3 years ago

ZhouCX117 commented 3 years ago

Hi, thank you for your excellent work~ It seems that the provided checkpoint is not right.

I run datasetGAN/test_deeplab_cross_validation.py, and the wrong is as follows: "File "datasetGAN/test_deeplab_cross_validation.py", line 357, in test classifier.load_state_dict(checkpoint['model_state_dict']) Missing key(s) in state_dict: "backbone.conv1.weight", "backbone.bn1.weight", "backbone.bn1.bias", "backbone.bn1.running_mean", "backbone.bn1.running_var".... Unexpected key(s) in state_dict: "module.layers.0.weight",.."

The parameters in the classifier and the provided .pth model are not consistent.

arieling commented 3 years ago

Hey we didn't provide deeplab checkpoints.

ZhouCX117 commented 3 years ago

Thanks for your reply！So I have to train one？ Is there a shortcut if I only want to have a try？ Do you provide a demo or colab？