Questions about batchnorm errors in Neural_Volume_Net

[pnpmpnp]

hello. First of all, thank you for sharing this great research.

1. However, the Neural_Volume_Net that I define in line 736 of my render_models.py file is the code to aggregate 5D Volumes. But when I run the actual code, I get an error because the batchnorm is declared as 2D. I think I need to change it to BatchNorm3d, what do you think?

class Neural_Volume_Net(nn.Module):
    def __init__(self,
         num_groups=1,
         norm_act=nn.BatchNorm2d,
         levels=1):

2. Also, I was wondering why you created an empty channel as much as C for volume_feature_no_ref with no reference image on line 83 of the casmvsnet.py file?

torch.empty((B, 3*(num_views-1) + C, num_depth, *ref_feature.shape[-2:]), device=imgs.device, dtype=torch.float)