Issue with visualizer

[sarthak247]

Heyaa!!
I was just playing around with your project and experimenting around with it. I wanted to visualize the latents and play around with them but it throws up the following error
Command: python visualizer.py

Error

Loading "https://api.ngc.nvidia.com/v2/models/nvidia/research/eg3d/versions/1/files/ffhq512-128.pkl"... Failed!

Traceback (most recent call last):
  File "/home/sarthak/Desktop/Portrait-4D/portrait4d/viz/renderer.py", line 165, in get_network
    data = legacy.load_network_pkl(f)
  File "/home/sarthak/Desktop/Portrait-4D/portrait4d/legacy.py", line 25, in load_network_pkl
    data = _LegacyUnpickler(f).load()
  File "/home/sarthak/Desktop/Portrait-4D/portrait4d/torch_utils/persistence.py", line 192, in _reconstruct_persistent_obj
    module = _src_to_module(meta.module_src)
  File "/home/sarthak/Desktop/Portrait-4D/portrait4d/torch_utils/persistence.py", line 228, in _src_to_module
    exec(src, module.__dict__) # pylint: disable=exec-used
  File "<string>", line 2, in <module>
ModuleNotFoundError: No module named 'training.networks_stylegan2'

Any idea what could be going wrong?