error while trying to run colab notebook: Nerfies Training v2.ipynb

[corlangerak]

HI I was trying to run the colab notebook Nerfies Training v2.ipynb after having run the Nerfies Capture Processing v2.ipynb on a video.

I got stuck on an error that I don't understand due to my lack of knowledge. maybe someone can help. This is the error the notebook produced

UnfilteredStackTrace Traceback (most recent call last) in () 34 use_warp_jacobian=train_config.use_elastic_loss, ---> 35 use_weights=train_config.use_elastic_loss) 36

26 frames UnfilteredStackTrace: TypeError: broadcast_to requires ndarray or scalar arguments, got <class 'list'> at position 0.

The stack trace below excludes JAX-internal frames. The preceding is the original exception that occurred, unmodified.

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

TypeError Traceback (most recent call last) /usr/local/lib/python3.7/dist-packages/jax/_src/numpy/util.py in _check_arraylike(fun_name, *args) 293 if not _arraylike(arg)) 294 msg = "{} requires ndarray or scalar arguments, got {} at position {}." --> 295 raise TypeError(msg.format(fun_name, type(arg), pos)) 296 297

TypeError: broadcast_to requires ndarray or scalar arguments, got <class 'list'> at position 0.

Thanks in advanced

[wuzijian1997]

I met the same question. Maybe you should try

!pip install flax==0.3.4 immutabledict mediapy !pip install jax==0.2.20 !pip install git+https://github.com/google/nerfies@v2

at the beginning to set the environment.

[sjtuytc]

I met the same question. Maybe you should try

!pip install flax==0.3.4 immutabledict mediapy !pip install jax==0.2.20 !pip install git+https://github.com/google/nerfies@v2

at the beginning to set the environment.

Not solved by this. Another issue was reported (import jax failed).

[CurtinComputing]

I met the same error

[pscore30]

I have also the same error!!