Source and weight input channels mismatch

I set the device to ‘mps’
but it shows:

/AppleInternal/Library/BuildRoots/b6051351-c030-11ec-96e9-3e7866fcf3a1/Library/Caches/com.apple.xbs/Sources/MetalPerformanceShadersGraph/mpsgraph/MetalPerformanceShadersGraph/Runtimes/MPSRuntime/Operations/GPUConv2DOps.mm:214: failed assertion `Source and weight input channels mismatch'

and then the program quit.

it also continuous showed me that:

anaconda3/envs/Unet/lib/python3.8/site-packages/torch/amp/autocast_mode.py:198: UserWarning: User provided device_type of 'cuda', but CUDA is not available. Disabling
  warnings.warn('User provided device_type of \'cuda\', but CUDA is not available. Disabling')

What’s wrong with this. How can I solve it. Thank you.

/AppleInternal/Library/BuildRoots/b6051351-c030-11ec-96e9-3e7866fcf3a1/Library/Caches/com.apple.xbs/Sources/MetalPerformanceShadersGraph/mpsgraph/MetalPerformanceShadersGraph/Runtimes/MPSRuntime/Operations/GPUConv2DOps.mm:214: failed assertion `Source and weight input channels mismatch’

Hi,

Are you using AMP? It does not support mps right now I’m afraid (We didn’t had time to work on that yet).
Also the channel mismatch error seems unrelated to mps. Do you get the same thing if you run on CPU?

Yes, I’m using the amp. Really thankful for your contribution.
Maybe there’s something with my code, I didn’t get the mismatch error when I used the device “cpu”

Can you try to run without AMP on MPS and see if that works?