force input to complex conv to have the same dtype as the weights, in the...
force input to complex conv to have the same dtype as the weights, in the hacked class to make sure complex valued network can do distributed training
Loading
Please register or sign in to comment