Error when trying to use Tensorflow mixed-precision on M1 Pro

Graphics & Games Metal
naor-navina OP
Created Oct ’22
Replies 1
Boosts 0
Views 2.1k
Participants 1

I tried training my model on my M1 Pro using Tensorflow's mixed-precision, hoping it will boost the performance, but I got an error: .../mpsgraph/MetalPerformanceShadersGraph/Core/Files/MPSGraphUtilities.mm:289:0: error: 'mps.select' op failed to verify that all of {true_value, false_value, result} have same element type

.../mpsgraph/MetalPerformanceShadersGraph/Core/Files/MPSGraphUtilities.mm:289:0: note: see current operation: %5 = "mps.select"(%4, %3, %2) : (tensor<1xi1>, tensor<1xf16>, tensor<1xf32>) -> tensor<1xf16>

Replies  1
Boosts  0
Views  2.1k
Participants  1
naor-navina OP
Oct ’22

It might worth adding that I'm fine tuning a model that was originally trained in fp32, but I think that shouldn't be a problem. Let me know if that assumption is incorrect and if so, what I need to change for it to work.

0
Error when trying to use Tensorflow mixed-precision on M1 Pro
First post date Last post date
 
 
Q