Daniel Tisdale

Results 1 comments of Daniel Tisdale

Modifying: https://github.com/google/hypernerf/blob/main/hypernerf/model_utils.py#L114 From: > jnp.broadcast_to([last_sample_z], z_vals[..., :1].shape) To: > import numpy as np > ... > jnp.broadcast_to(np.asarray([last_sample_z]), z_vals[..., :1].shape) Fixed the problem for me. (jax version fixed to jax==0.2.20 jaxlib==0.1.71)