It’s effective, and many thanks! But there is also a quesiton: the error emerged in my function ‘_fun(input,weight, bias)’; then I print the input shape, and found that the data has been split relative to number of GPUs; and I also found that the bias could be None in Conv module, so why couldn’t the bias be None here?