Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

pretrained/CelebA-HQ_512.pkl _pickle.UnpicklingError: pickle data was truncated #120

Open
136LYJ opened this issue Jul 19, 2024 · 0 comments

Comments

@136LYJ
Copy link

136LYJ commented Jul 19, 2024

python generate_image.py --network pretrained/CelebA-HQ_512.pkl --dpath /home/data/PersonalData/lianyj/erase/cut/img4/img_expand-re512_rgb --mpath /home/data/PersonalData/lianyj/ENet-LGNet/mask/erase-rule-expand-re512-inv --outdir /home/data/PersonalData/lianyj/MAT/m-01
Loading data from: /home/data/PersonalData/lianyj/erase/cut/img4/img_expand-re512_rgb
Loading mask from: /home/data/PersonalData/lianyj/ENet-LGNet/mask/erase-rule-expand-re512-inv
Loading networks from: pretrained/CelebA-HQ_512.pkl
Traceback (most recent call last):
File "generate_image.py", line 157, in
generate_images() # pylint: disable=no-value-for-parameter
File "/home/lianyj/miniconda3/envs/MAT/lib/python3.7/site-packages/click/core.py", line 1157, in call
return self.main(*args, **kwargs)
File "/home/lianyj/miniconda3/envs/MAT/lib/python3.7/site-packages/click/core.py", line 1078, in main
rv = self.invoke(ctx)
File "/home/lianyj/miniconda3/envs/MAT/lib/python3.7/site-packages/click/core.py", line 1434, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "/home/lianyj/miniconda3/envs/MAT/lib/python3.7/site-packages/click/core.py", line 783, in invoke
return _callback(*args, **kwargs)
File "/home/lianyj/miniconda3/envs/MAT/lib/python3.7/site-packages/click/decorators.py", line 33, in new_func
return f(get_current_context(), *args, **kwargs)
File "generate_image.py", line 100, in generate_images
G_saved = legacy.load_network_pkl(f)['G_ema'].to(device).eval().requires_grad
(False) # type: ignore
File "/home/lianyj/MAT/legacy.py", line 21, in load_network_pkl
data = _LegacyUnpickler(f).load()
_pickle.UnpicklingError: pickle data was truncated

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant