I had another problem,too.
#conv_phi
import torch
import numpy as np
import torch.nn as nn
B,h,w,Sr,b,n=16,32,56,0.1,8,25
kernel_size=(1,1,b*b)
x=torch.rand(B,b*b,h,w)
print(x.shape)
conv=nn.Conv3d(b*b,int(np.ceil(0.1*b*b)),kernel_size)
print(conv)
conv(x).shapepe or paste code here

error:
Expected 5-dimensional input for 5-dimensional weight 7 64 1 1 64, but got 4-dimensional input of size [16, 64, 32, 56] instead

I will very thankful if anybody helps me solve this error and explains how to encounter with such errors.