white_hair.pt is not a state_dict #4
Description
Thank you for providing the code. But I fail to run the python inference.py --config ./configs/white_hair.yml --work_dir work_dirs/white_hair/ in README.md. I've clicked the "white hair" link in README and downloaded the complete "celaba_100_w+_codes.pt" to the ./pretrain folder then renamed it to "white_hair.pt". And when I run the inference I just receive:
Did I download the wrong "white_hair.pt"? It seems like the weights are not a state_dict but a Tensor.
Can you please give a solution to this ? Thank you.