datasetGAN_release icon indicating copy to clipboard operation
datasetGAN_release copied to clipboard

checkpoint

Open ZhouCX117 opened this issue 3 years ago • 2 comments

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.

ZhouCX117 avatar Jul 07 '21 11:07 ZhouCX117

Hey we didn't provide deeplab checkpoints.

arieling avatar Jul 07 '21 16:07 arieling

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?

ZhouCX117 avatar Jul 08 '21 01:07 ZhouCX117