UnboundLocalError: cannot access local variable 'reset_w' where it is not associated with a value

Question

UnboundLocalError: cannot access local variable 'reset_w' where it is not associated with a value

p4ulor opened this issue a year ago · comments

Paulo R. commented a year ago

How this error occurred

$ sh scripts/gui.sh

What shows on the screen

The GUI appears
It shows the text saying something like "loading lion module"
Shows some lines of code lines in place of the lion image
Shows "dragcan is not responding"
Closes and shows this output

Outputs

Loading "/home/paulor/Public/Pulls/DragGAN/checkpoints/stylegan2_lions_512_pytorch.pkl"... Done.
()
{'z_dim': 512, 'c_dim': 0, 'w_dim': 512, 'img_resolution': 512, 'img_channels': 3, 'mapping_kwargs': {'num_layers': 8, 'embed_features': None, 'layer_features': None, 'activation': 'lrelu', 'lr_multiplier': 0.01, 'w_avg_beta': 0.995}, 'synthesis_kwargs': {'channel_base': 32768, 'channel_max': 512, 'num_fp16_res': 0, 'conv_clamp': None, 'architecture': 'skip', 'resample_filter': [1, 3, 3, 1], 'use_noise': True, 'activation': 'lrelu'}}
Setting up PyTorch plugin "bias_act_plugin"... Failed!

Traceback (most recent call last):
  File "/home/paulor/miniconda3/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1893, in _run_ninja_build
    subprocess.run(
  File "/home/paulor/miniconda3/lib/python3.11/subprocess.py", line 571, in run
    raise CalledProcessError(retcode, process.args,
subprocess.CalledProcessError: Command '['ninja', '-v']' returned non-zero exit status 1.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/home/paulor/Public/Pulls/DragGAN/viz/renderer.py", line 112, in render
    self.init_network(res, **args)
  File "/home/paulor/Public/Pulls/DragGAN/viz/renderer.py", line 249, in init_network
    w = G.mapping(z, label, truncation_psi=trunc_psi, truncation_cutoff=trunc_cutoff)
        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/paulor/miniconda3/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/paulor/Public/Pulls/DragGAN/training/networks_stylegan2.py", line 247, in forward
    x = layer(x)
        ^^^^^^^^
  File "/home/paulor/miniconda3/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/paulor/Public/Pulls/DragGAN/training/networks_stylegan2.py", line 125, in forward
    x = bias_act.bias_act(x, b, act=self.activation)
        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/paulor/Public/Pulls/DragGAN/torch_utils/ops/bias_act.py", line 84, in bias_act
    if impl == 'cuda' and x.device.type == 'cuda' and _init():
                                                      ^^^^^^^
  File "/home/paulor/Public/Pulls/DragGAN/torch_utils/ops/bias_act.py", line 41, in _init
    _plugin = custom_ops.get_plugin(
              ^^^^^^^^^^^^^^^^^^^^^^
  File "/home/paulor/Public/Pulls/DragGAN/torch_utils/custom_ops.py", line 136, in get_plugin
    torch.utils.cpp_extension.load(name=module_name, build_directory=cached_build_dir,
  File "/home/paulor/miniconda3/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1284, in load
    return _jit_compile(
           ^^^^^^^^^^^^^
  File "/home/paulor/miniconda3/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1509, in _jit_compile
    _write_ninja_file_and_build_library(
  File "/home/paulor/miniconda3/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1624, in _write_ninja_file_and_build_library
    _run_ninja_build(
  File "/home/paulor/miniconda3/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1909, in _run_ninja_build
    raise RuntimeError(message) from e
RuntimeError: Error building extension 'bias_act_plugin': [1/3] /usr/bin/nvcc  -DTORCH_EXTENSION_NAME=bias_act_plugin -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -isystem /home/paulor/miniconda3/lib/python3.11/site-packages/torch/include -isystem /home/paulor/miniconda3/lib/python3.11/site-packages/torch/include/torch/csrc/api/include -isystem /home/paulor/miniconda3/lib/python3.11/site-packages/torch/include/TH -isystem /home/paulor/miniconda3/lib/python3.11/site-packages/torch/include/THC -isystem /home/paulor/miniconda3/include/python3.11 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_BFLOAT16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_86,code=compute_86 -gencode=arch=compute_86,code=sm_86 --compiler-options '-fPIC' --use_fast_math --allow-unsupported-compiler -std=c++17 -c /home/paulor/.cache/torch_extensions/py311_cu117/bias_act_plugin/3cb576a0039689487cfba59279dd6d46-nvidia-geforce-rtx-3050-ti-laptop-gpu/bias_act.cu -o bias_act.cuda.o 
FAILED: bias_act.cuda.o 
/usr/bin/nvcc  -DTORCH_EXTENSION_NAME=bias_act_plugin -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -isystem /home/paulor/miniconda3/lib/python3.11/site-packages/torch/include -isystem /home/paulor/miniconda3/lib/python3.11/site-packages/torch/include/torch/csrc/api/include -isystem /home/paulor/miniconda3/lib/python3.11/site-packages/torch/include/TH -isystem /home/paulor/miniconda3/lib/python3.11/site-packages/torch/include/THC -isystem /home/paulor/miniconda3/include/python3.11 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_BFLOAT16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_86,code=compute_86 -gencode=arch=compute_86,code=sm_86 --compiler-options '-fPIC' --use_fast_math --allow-unsupported-compiler -std=c++17 -c /home/paulor/.cache/torch_extensions/py311_cu117/bias_act_plugin/3cb576a0039689487cfba59279dd6d46-nvidia-geforce-rtx-3050-ti-laptop-gpu/bias_act.cu -o bias_act.cuda.o 
/usr/include/c++/11/bits/std_function.h:435:145: error: parameter packs not expanded with ‘...’:
  435 |         function(_Functor&& __f)
      |                                                                                                                                                 ^ 
/usr/include/c++/11/bits/std_function.h:435:145: note:         ‘_ArgTypes’
/usr/include/c++/11/bits/std_function.h:530:146: error: parameter packs not expanded with ‘...’:
  530 |         operator=(_Functor&& __f)
      |                                                                                                                                                  ^ 
/usr/include/c++/11/bits/std_function.h:530:146: note:         ‘_ArgTypes’
[2/3] c++ -MMD -MF bias_act.o.d -DTORCH_EXTENSION_NAME=bias_act_plugin -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -isystem /home/paulor/miniconda3/lib/python3.11/site-packages/torch/include -isystem /home/paulor/miniconda3/lib/python3.11/site-packages/torch/include/torch/csrc/api/include -isystem /home/paulor/miniconda3/lib/python3.11/site-packages/torch/include/TH -isystem /home/paulor/miniconda3/lib/python3.11/site-packages/torch/include/THC -isystem /home/paulor/miniconda3/include/python3.11 -D_GLIBCXX_USE_CXX11_ABI=0 -fPIC -std=c++17 -c /home/paulor/.cache/torch_extensions/py311_cu117/bias_act_plugin/3cb576a0039689487cfba59279dd6d46-nvidia-geforce-rtx-3050-ti-laptop-gpu/bias_act.cpp -o bias_act.o 
ninja: build stopped: subcommand failed.



Traceback (most recent call last):
  File "/home/paulor/Public/Pulls/DragGAN/visualizer_drag.py", line 403, in <module>
    main()
  File "/home/paulor/miniconda3/lib/python3.11/site-packages/click/core.py", line 1157, in __call__
    return self.main(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/paulor/miniconda3/lib/python3.11/site-packages/click/core.py", line 1078, in main
    rv = self.invoke(ctx)
         ^^^^^^^^^^^^^^^^
  File "/home/paulor/miniconda3/lib/python3.11/site-packages/click/core.py", line 1434, in invoke
    return ctx.invoke(self.callback, **ctx.params)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/paulor/miniconda3/lib/python3.11/site-packages/click/core.py", line 783, in invoke
    return __callback(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/paulor/Public/Pulls/DragGAN/visualizer_drag.py", line 397, in main
    viz.draw_frame()
  File "/home/paulor/Public/Pulls/DragGAN/visualizer_drag.py", line 156, in draw_frame
    self.latent_widget(expanded)
  File "/home/paulor/Public/Pulls/DragGAN/gui_utils/imgui_utils.py", line 83, in decorator
    res = method(self, *args, **kwargs)
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/paulor/Public/Pulls/DragGAN/viz/latent_widget.py", line 93, in __call__
    viz.args.reset_w = reset_w
                       ^^^^^^^
UnboundLocalError: cannot access local variable 'reset_w' where it is not associated with a value
(base)

My system

conda 23.5.2
Python 3.11.4
Ubuntu 22.04.2 LTS
11th Gen Intel® Core™ i7-11850H @ 2.50GHz
NVIDIA Corporation GA107M [GeForce RTX 3050 Ti Mobile] (w/ Nvidia driver metapackage from nvidia-driver-535 (proprietary, tested)