model dimension settings.

Hi, this project is excellent work! But i am kind of confused about how to set the model dimension.For example, when my data's config like [batchsize,channels,depth,height,width], and i try to use resnet50 for example, but the problem shows up: RuntimeError: Expected 3D (unbatched) or 4D (batched) input to conv2d, but got input of size: [4, 8, 14, 112, 112]. So are there any chances to set models' dimension? Maybe you already solve this problem but could you tell me how? I would be thankful if you could reply me soon.