yunjey / stargan

StarGAN - Official PyTorch Implementation (CVPR 2018)
MIT License
5.23k stars 970 forks source link

Error when testing #124

Open Hcz-up opened 4 years ago

Hcz-up commented 4 years ago

question: when i input:python main.py --mode test --dataset CelebA --image_size 128 --c_dim 5 --selected_attrs Black_Hair Blond_Hair Brown_Hair Male Young --model_save_dir=stargan_celeba\models --result_dir=stargan_celeba\results have a traceback: Traceback (most recent call last): File "main.py", line 115, in main(config) File "main.py", line 49, in main solver.test() File "C:\Users\HCZ\Desktop\stargan-master\solver.py", line 530, in test self.restore_model(self.test_iters) File "C:\Users\HCZ\Desktop\stargan-master\solver.py", line 107, in restore_model self.G.load_state_dict(torch.load(G_path, map_location=lambda storage, loc:storage)) File "E:\language\python\lib\site-packages\torch\serialization.py", line 593, in load return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args) File "E:\language\python\lib\site-packages\torch\serialization.py", line 763, in _legacy_load magic_number = pickle_module.load(f, pickle_load_args) _pickle.UnpicklingError: invalid load key, '?'.