Shape pre-trained stage error about nerfactor HOT 2 CLOSED

[60, 497] does not index into param shape [256,341,3]

Looks like a shape mismatch to me. I recommend debugging with the non-distributed version so that you can insert a breakpoint there and inspect the tensor shape. Feel free to reopen this if the issue persists.

from nerfactor.

[60, 497] does not index into param shape [256,341,3]

Looks like a shape mismatch to me. I recommend debugging with the non-distributed version so that you can insert a breakpoint there and inspect the tensor shape. Feel free to reopen this if the issue persists.

It turns out to be my mistake. I think I may wrongly set the imh as 512 when generating the dataset.

# Resize
       if imh != xyz.shape[0]:
           xyz = xm.img.resize(xyz, new_h=imh)
           normal = xm.img.resize(normal, new_h=imh)
           lvis = xm.img.resize(lvis, new_h=imh)
           alpha = xm.img.resize(alpha, new_h=imh)
           rgb = xm.img.resize(rgb, new_h=imh)
       rgb = xm.img.resize(rgb, new_h=imh)

After I forced the rgb to resize everything went okay. Thanks anyway!

from nerfactor.