Open Marquess98 opened 1 year ago
I think set eval = False can solve it
Sorry to bother. There is another problem, the error reporting are as follows. looking forward to your reply, thank you!
I1013 17:02:48.312762 140074248000064 checkpoints.py:249] Found no checkpoint files in /home/bran/GeCoNeRF-main/geco_train/orchids
200/85000: i_loss=0.2004, rgb_loss=0.0000, reg_loss=0.0000, dec_reg_loss=0.0000, avg_loss=0.2282, weight_l2=0.00e+00, lr=6.63e-05, 1700957 rays/sec
400/85000: i_loss=0.1800, rgb_loss=0.0000, reg_loss=0.0000, dec_reg_loss=0.0000, avg_loss=0.1830, weight_l2=0.00e+00, lr=1.25e-04, 1999249 rays/sec
600/85000: i_loss=0.1679, rgb_loss=0.0000, reg_loss=0.0000, dec_reg_loss=0.0000, avg_loss=0.1675, weight_l2=0.00e+00, lr=1.81e-04, 1518354 rays/sec
800/85000: i_loss=0.1484, rgb_loss=0.0000, reg_loss=0.0000, dec_reg_loss=0.0000, avg_loss=0.1586, weight_l2=0.00e+00, lr=2.33e-04, 1535362 rays/sec
Traceback (most recent call last):
File "train.py", line 488, in
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:
Traceback (most recent call last):
File "train.py", line 488, in
@Marquess98 Hi! I have the same problem when I run the codes. Have you fixed this?
Hi,thanks for your work! Here I’ve encountered some problems when run your code. I think there are some mistakes in reprojector function in match.py file. looking forward to your update! Thank you! Cannot divide evenly the sizes of shapes (3, 16) and (14400, 16) File "/home/bran/GeCoNeRF-main/train.py", line 488, in
app.run(main)
File "/home/bran/GeCoNeRF-main/train.py", line 330, in main
w_grad, outputs, keys = reg_utils.reg_train_step(render_trainfn, reg_model, dataset_type, 120, reg_type, keys[0], state, reg_params, reg_batch, alpha_batch, step)
File "/home/bran/GeCoNeRF-main/internal/reg_utils.py", line 155, in reg_train_step
(loss , outputs), grad = jax.value_and_grad(warp_loss_fn, has_aux = True)(jax.device_get(jax.tree_map(lambda x:x[0], state)).optimizer.target)
File "/home/bran/GeCoNeRF-main/internal/reg_utils.py", line 93, in warp_loss_fn
data_type = data_type
File "/home/bran/GeCoNeRF-main/internal/models.py", line 377, in render_image
rng, data_type, eval = eval)
File "/home/bran/GeCoNeRF-main/internal/match.py", line 127, in reprojector
pose = pose.reshape(rays.pose.shape),
The above exception was the direct cause of the following exception:
File "/home/bran/GeCoNeRF-main/internal/match.py", line 127, in reprojector pose = pose.reshape(rays.pose.shape), File "/home/bran/GeCoNeRF-main/internal/models.py", line 377, in render_image rng, data_type, eval = eval) File "/home/bran/GeCoNeRF-main/internal/reg_utils.py", line 93, in warp_loss_fn data_type = data_type File "/home/bran/GeCoNeRF-main/internal/reg_utils.py", line 155, in reg_train_step (loss , outputs), grad = jax.value_and_grad(warp_loss_fn, has_aux = True)(jax.device_get(jax.tree_map(lambda x:x[0], state)).optimizer.target) File "/home/bran/GeCoNeRF-main/train.py", line 330, in main w_grad, outputs, keys = reg_utils.reg_train_step(render_trainfn, reg_model, dataset_type, 120, reg_type, keys[0], state, reg_params, reg_batch, alpha_batch, step) File "/home/bran/GeCoNeRF-main/train.py", line 488, in
app.run(main)