Error with 3D Shape reconstruction

[Sentient07]

Hello,

Congrats on this amazing work and thanks for open-sourcing the code. I am attempting to run the notebook for 3D shape reconstruction locally and I face the error attached below,

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-9-8f4605d1a63b> in <module>
     32         print(expname)
     33 
---> 34         out = run_training(embed_params, mesh, corners, test_pts, render_args_lr, expname)
     35         tests_all[expname] = out[1]
     36         out_all[expname] = out

<ipython-input-5-b42092951119> in run_training(embed_params, mesh, corners, test_pts, render_args_lr, name)
     81             xs.append(i)
     82             slices = [outputs, pred, np.abs(pred - outputs)]
---> 83             renderings = list(render_rays(get_params(opt_state), ab, *render_args_lr))
     84             renderings.append(make_normals(render_args_lr[0], renderings[0]) * .5 + .5)
     85 

ValueError: Non-hashable static arguments are not supported. An error occured while trying to hash an object of type <class 'list'>, [TrackedArray([-0.001     ,  0.24340938,  0.20585178]), TrackedArray([0.90775315, 0.74892081, 0.81103368])]. The error was:
TypeError: unhashable type: 'list'

Could this be due to version mismatch in JAX? I'm using jax on GPU with Cuda 11.3. Also, along with the package requirement, can you also please upload an environment.yml or requirement.txt if possible? Thanks

[nexuslrf]

I also encountered this error. I am running the latest version of JAX.
After doing some debugging, I found that the problem lies in the load_mesh() function in the 2nd cell.
You can change corners = [c0, c1] into corners = (c0, c1), then this error will disappear.
I think the reason is that List type is non-static in Python compared to Tuple type.