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

[zijianwu1231]

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.

[Zelin Zhao]

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!!