TypeError: expected size to be one of int or Tuple[int] or Tuple[int, int] or Tuple[int, int, int], but got size with types [<class 'numpy.int64'>, <class 'numpy.int64'>]

[liuhw01]

When I load the network from pretrained_models/next3d_ffhq_512.pkl,

Running img = G.synthesis() in all the code results in an error:

_ = G.synthesis(ws[:1], c[:1], verts[:1]) # warm up

File "", line 154, in synthesis
File "/root/miniconda3/envs/next3d/lib/python3.9/site-packages/torch/nn/functional.py", line 3924, in interpolate
raise TypeError(
TypeError: expected size to be one of int or Tuple[int] or Tuple[int, int] or Tuple[int, int, int], but got size with types [<class 'numpy.int64'>, <class 'numpy.int64'>]

What could be the reason for this?

[zjh21]

same problem here

[mhwu2017]

I got the same problem, and solved it by downgrading the torch version from 2.1.0 to 1.14.1