Post

Replies

Boosts

Views

Created

Does the conv3d layer of CoreML support the GPU mode?
A CoreML model with conv3d layers runs very slow on my mac. If I set usesCPUOnly to True, it will take about the same running time. So it seems that the conv3d layer only supports the CPU mode in a CoreML. When I replace conv3d layers with conv2d layers, the model will be many times faster than before. The macOS version is 11.2.3. The CoreML model is converted from a pytorch model.
1
0
936
Dec ’21