I have some question in gen_image.py

[wj7486]

Traceback (most recent call last):
File "gen_images.py", line 143, in
generate_images() # pylint: disable=no-value-for-parameter
File "/home/wangjie/anaconda3/envs/torch18/lib/python3.7/site-packages/click/core.py", line 1128, in call
return self.main(*args, **kwargs)
File "/home/wangjie/anaconda3/envs/torch18/lib/python3.7/site-packages/click/core.py", line 1053, in main
rv = self.invoke(ctx)
File "/home/wangjie/anaconda3/envs/torch18/lib/python3.7/site-packages/click/core.py", line 1395, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "/home/wangjie/anaconda3/envs/torch18/lib/python3.7/site-packages/click/core.py", line 754, in invoke
return __callback(*args, **kwargs)
File "gen_images.py", line 108, in generate_images
G = legacy.load_network_pkl(f)['G_ema'].to(device) # type: ignore
File "/home/wangjie/workspace/projected-gan-main/legacy.py", line 24, in load_network_pkl
data = _LegacyUnpickler(f).load()
File "/home/wangjie/workspace/projected-gan-main/legacy.py", line 69, in find_class
return super().find_class(module, name)
ModuleNotFoundError: No module named 'builtin'
(torch18) wangjie@vm02:~/workspace/projected-gan-main$ python gen_images.py --outdir=out --trunc=0.7 --seeds=600-605 --class=99 --network=./training-runs/00000-fastgan-CELEBA500_unlabeledall-gpus4-batch64/best_model.pkl
Loading networks from "./training-runs/00000-fastgan-CELEBA500_unlabeledall-gpus4-batch64/best_model.pkl"...
Traceback (most recent call last):
File "gen_images.py", line 143, in
generate_images() # pylint: disable=no-value-for-parameter
File "/home/wangjie/anaconda3/envs/torch18/lib/python3.7/site-packages/click/core.py", line 1128, in call
return self.main(*args, **kwargs)
File "/home/wangjie/anaconda3/envs/torch18/lib/python3.7/site-packages/click/core.py", line 1053, in main
rv = self.invoke(ctx)
File "/home/wangjie/anaconda3/envs/torch18/lib/python3.7/site-packages/click/core.py", line 1395, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "/home/wangjie/anaconda3/envs/torch18/lib/python3.7/site-packages/click/core.py", line 754, in invoke
return __callback(*args, **kwargs)
File "gen_images.py", line 108, in generate_images
G = legacy.load_network_pkl(f)['G_ema'].to(device) # type: ignore
File "/home/wangjie/workspace/projected-gan-main/legacy.py", line 24, in load_network_pkl
data = _LegacyUnpickler(f).load()
File "/home/wangjie/workspace/projected-gan-main/legacy.py", line 72, in find_class
return super().find_class(module, name)
ModuleNotFoundError: No module named 'builtin'

I don't know how to deal.Can you help me

[anaivebird]

add this to line 72 of legacy.py