dimension error in gcnetwork

[haojeng-wang]

Hi,
I've download your code and would like to do testing first.
When running test.py, I got the error
"ValueError: The channel dimension of the inputs should be defined. Found None."

I run it by the argument list: -data ./data -pspath ./output -wpath ./model/shared_cost_weight.hdf5
I found
cv = Lambda(getCostVolume, arguments = {'max_d':int(d/2)}, output_shape = (int(d/2), None, None, num_filters * 2))(unifeatures)
takes several arguments but the function tooks only the max_d instead of all the parameters.
how could I solve this, thanks!