MPSCNNSoftMaxNode's result image format?!

Question

Created Mar ’20

Replies 1

Boosts 0

Participants 1

I am using a neural network with a data type MPSDataTypeFloat32 but it seems to be that the final MPSCNNSoftMaxNode for the inference run returns an image using MPSDataTypeFloat16. Is this always the case? How do the nodes determine the output data type? Can I change them?

Answered by Hardy in 411053022

I am going to answer my own question: when using graphs (of MPSNNGraph type), setting the graph's format to MPSImageFeatureChannelFormatFloat32 or to any other suitable type does the job.

Boost

Answer 1

Hardy OP

Mar ’20

Accepted Answer

I am going to answer my own question: when using graphs (of MPSNNGraph type), setting the graph's format to MPSImageFeatureChannelFormatFloat32 or to any other suitable type does the job.

0