harskish / ganspace

Discovering Interpretable GAN Controls [NeurIPS 2020]
Apache License 2.0
1.78k stars 264 forks source link

TypeError: forward() got an unexpected keyword argument 'input_is_w' #32

Closed chengkeng closed 3 years ago

chengkeng commented 3 years ago

Hi! First of all, thank you very much for your contribution!


Only this code can work smoothly:

Explore BigGAN-deep husky python interactive.py --model=BigGAN-512 --class=husky --layer=generator.gen_z -n=1_000_000



The following prompt appears when other codes are executed:

TypeError: forward() got an unexpected keyword argument 'input_is_w'



Traceback (most recent call last): File "interactive.py", line 645, in setup_model() File "interactive.py", line 144, in setup_model inst = get_instrumented_model(model_name, class_name, layer_name, torch.device('cuda'), use_w=args.use_w) File "C:\Users\Creator\miniconda3\envs\ganspace\lib\functools.py", line 840, in wrapper return dispatch(args[0].class)(*args, kw) File "C:\MyWork\My_GAN_Work\ganspace-master\models\wrappers.py", line 723, in get_instrumented_model latent_shape = model.get_latent_shape() File "C:\MyWork\My_GAN_Work\ganspace-master\netdissect\modelconfig.py", line 107, in create_instrumented_model latent_shape=getattr(args, 'latent_shape', None)) File "C:\MyWork\My_GAN_Work\ganspace-master\netdissect\modelconfig.py", line 137, in annotate_model_shapes output = model(dry_run) File "C:\Users\Creator\miniconda3\envs\ganspace\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl result = self.forward(*input, *kwargs) File "C:\MyWork\My_GAN_Work\ganspace-master\netdissect\nethook.py", line 48, in forward return self.model(inputs, kwargs) File "C:\Users\Creator\miniconda3\envs\ganspace\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl result = self.forward(*input, *kwargs) File "C:\MyWork\My_GAN_Work\ganspace-master\models\wrappers.py", line 191, in forward truncation=self.truncation, truncation_latent=self.latent_avg, input_is_w=self.w_primary) File "C:\Users\Creator\miniconda3\envs\ganspace\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl result = self.forward(input, **kwargs) TypeError: forward() got an unexpected keyword argument 'input_is_w'



This is my conda list:

Name Version Build Channel

_tflowselect____ 2.3.0mkl____main absl-py0.11.0____pyhd3eb1b0_1main aim __ 2.1.5_____ pypi_0pypi aimrecords0.0.7____pypi_0pypi anytree___ 2.8.0_ pypi_0pypi appdirs____1.4.4pypi0pypi astor ____ 0.8.1 __py37_0main base58____2.0.1___ pypi_0pypi blas ____ 1.0 ____ mklmain blosc _____1.20.1__h7bd577a_0main boto31.16.55pyhd3eb1b0_0 botocore____1.19.55__pyhd3eb1b0_0main brotli ____1.0.9_____ha925a31_2main brotlipy 0.7.0____py37h2bbff1b_1003main bzip21.0.8he774522_0main ca-certificates__2020.12.8_____haa95532_0main cached-property____1.5.2___pypi_0pypi certifi____2020.12.5___pypi_0pypi cffi__1.14.4____py37hcd4344a_0main chardet____4.0.0py37haa95532_1003main charls___2.1.0_h33f27b4_2main click__7.1.2____pyhd3eb1b0_0main cloudpickle_1.6.0___py_0main cmake____3.18.2_____hab937c2_0main cryptography_3.3.1___py37hcd4344a_0main cudatoolkit____10.1.243h74a9793_0main cycler__0.10.0___py37_0main cython___0.29.21__pypi_0pypi cytoolz____0.11.0____py37he774522_0main dask-core____2020.12.0__pyhd3eb1b0_0main decorator____4.4.2____py_0main deprecated____1.2.10____py_0main distro__1.5.0__pyhd3eb1b0_1 main docker__4.4.1____pypi_0pypi docutils____0.15.2____py37_0main fbpca____1.0__pypi_0pypi ffmpeg__4.2.2____he774522_0 fire__0.4.0____pypi_0pypi freetype____2.10.4__hd328e21_0main future__0.18.2__pypi_0pypi gast____0.2.2____pypi_0pypi giflib____5.2.1__h62dcd97_0main git__2.23.0__h6bb4b03_0main gitdb____4.0.5____pypi_0pypi gitpython__3.1.12____pypi_0pypi glumpy____1.1.0____pypi_0pypi grpcio__1.31.0____py37he7da953_0main h5py__3.1.0__pypi_0pypi hdf5__1.10.2____hac2f561_1main helpdev____0.7.1____pypi_0pypi icc_rt____2019.0.0__h0cc432a_1main icu__58.2__ha925a31_3main idna__2.10__pyhd3eb1b0_0main imagecodecs__2020.5.30____py37hb1be65f_2main imageio____2.9.0__py_0main importlib-metadata____3.4.0__pypi_0pypi intel-openmp__2021.1.2____pypi_0pypi jmespath__0.10.0____py_0main joblib__1.0.0____pyhd3eb1b0_0main jpeg__9b____hb83a4c4_2main keras-applications__1.0.8____py_1main keras-preprocessing____1.1.0____py_1main kiwisolver____1.3.0__py37hd77b12b_0main kornia____0.4.1____pypi_0pypi lcms2__2.11__hc51a39a_0main libaec__1.0.4____h33f27b4_1main libmklml____2019.0.5____haa95532_0main libpng__1.6.37__h2a8f88b_0main libprotobuf____3.13.0.1____h200bbdf_0main libtiff____4.1.0__h56a325e_1main libuv____1.39.0____he774522_0main libzopfli_____1.0.3____ha925a31_0main linear-attention-transformer0.15.3__pypi_0pypi linformer__0.2.1__pypi_0pypi local-attention__1.2.1____pypi_0pypi lz4-c____1.9.2____hf4a77e7_3main mako__1.1.4__pypi_0pypi markdown____3.3.3__py37haa95532_0main markupsafe__2.0.0a1____pypi_0pypi matplotlib____3.3.3__pypi_0pypi matplotlib-base__3.3.2__py37hba9282a_0main mkl____2020.2__256main mkl-service__2.3.0____py37h196d8e1_0main mkl_fft__1.2.0__py37h45dec08_0main mkl_random____1.1.1__py37h47e9c7a_0main networkx____2.5____py_0main ninja__1.10.0.post2____pypi_0pypi nltk__3.5____py_0 numpy____1.19.2__py37hadc3359_0main numpy-base__1.19.2__py37ha3acd2a_0main olefile__0.46____py37_0main opencv-python____4.5.1.48____pypi_0pypi openjpeg__2.3.0__h5ec785f_1main openssl____1.1.1i____h2bbff1b_0main opt-einsum____3.3.0____pypi_0pypi packaging_____20.8____pyhd3eb1b0_0 main pandas____1.2.0__py37hf11a4ad_0main pillow__6.2.1__py37hdc69c19_0 pip____20.3.3__py37haa95532_0 protobuf__3.14.0__pypi_0pypi psutil__5.8.0____pypi_0pypi py____1.10.0____pypi_0pypi pycparser__2.20____py_2main pycuda____2019.1.2+cuda101__pypi_0pypi pyopengl__3.1.4___pypi_0pypi pyopengltk____0.0.3____pypi_0 pypi pyopenssl__20.0.1____pyhd3eb1b0_1main pyparsing__2.4.7____pyhd3eb1b0_0main pyqt____5.9.2____py37h6538335_2main pyqt5____5.15.2____pypi_0pypi pyqt5-sip____12.8.1____pypi_0pypi pyrser____0.2.0____pypi_0pypi pysocks____1.7.1____py37_1main python____3.7.9____h60c2a47_0 python-dateutil____2.8.1____py_0main pytools____2021.1____pypi_0_pypi pytorch____1.6.0____py3.7_cuda101_cudnn7_0__pytorch pytz____2020.5____pyhd3eb1b0_0main pywavelets____1.1.1____py37he774522_2main pywin32____227____pypi_0pypi pyyaml____5.3.1____py37he774522_1main qdarkstyle____2.8.1____pypi_0pypi qt____5.9.7____vc14h73c81de_0main qtpy____1.9.0____pypi_0pypi regex____2020.11.13_____py37h2bbff1b_0main requests____2.25.1_pyhd3eb1b0_0 retry__0.9.2____pypi_0 pypi s3transfer____0.3.4____pyhd3eb1b0_0main scikit-build____0.11.1____py37hd77b12b_2main scikit-image____0.17.2____py37h1e1f486_0 scikit-learn____0.23.2____py37h47e9c7a_0 scipy__1.5.2____py37h9439919_0main setuptools____51.3.3____py37haa95532_4main sip____4.19.8____py37h6538335_0main six____1.15.0____py37haa95532_0main smmap____3.0.5____pypi_0pypi snappy____1.1.8____h33f27b4_0main sqlite____3.33.0____h2a8f88b_0main tensorboard____1.15.0____pypi_0pypi tensorflow____1.14.0____mkl_py37h7908ca0_0 main tensorflow-base____1.14.0____mkl_py37ha978198_0 main tensorflow-cpu-estimator____1.15.1____pypi_0pypi tensorflow-estimator____1.14.0____py_0main termcolor____1.1.0____pypi_0pypi threadpoolctl____2.1.0____pyh5ca1d4c_0main tifffile____2021.1.14____pyhd3eb1b0_1main tk____8.6.10____he774522_0main toolz____0.11.1____pyhd3eb1b0_0main torch____1.6.0+cu101____pypi_0pypi torchdiffeq____0.0.1____pypi_0pypi torchvision____0.1.8____pypi_0pypi tornado____6.1____py37h2bbff1b_0 main tqdm____4.55.1____pyhd3eb1b0_0 triangle____20190115.3____pypi_0pypi typing-extensions____3.7.4.3____pypi_0pypi urllib3____1.25.11____py_0main vc____14.2____h21ff451_1 main vs2015_runtime____14.27.29016____h5e58377_2main websocket-client__0.57.0__pypi_0pypi werkzeug__1.0.1__py_0main wheel__0.36.2____pyhd3eb1b0_0main win_inet_pton____1.1.0__py37haa95532_0main wincertstore__0.2__py37_0main wrapt____1.12.1____py37he774522_1main xz____5.2.5____h62dcd97_0main yaml__0.2.5__he774522_0main zipp____3.4.0____pypi_0pypi zlib____1.2.11__h62dcd97_4main zstd__1.4.8.1____pypi_0____pypi