checkpoint

[ZhouCX117]

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]

Hey we didn't provide deeplab checkpoints.

[ZhouCX117]

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？