Using the drawing classification model in iOS

To fix the previous RefSwitch error, which will occur no matter whether you use the TensorFlow Pod as we did in Chapter 2, Classifying Images with Transfer Learning, and Chapter 6, Describing Images in Natural Language, or the manually built TensorFlow library, as in the other chapters, we have to use some new trick. The error occurs because the INT64 data type is required for a RefSwitch operation, but it's not one of the registered data types built into the TensorFlow library because, by default, to make the library as small as possible, only common data types for each operation are included. We may fix this from the model building end in Python, but here we'll just show you how to fix this from the iOS end, which can be useful when you don't have access to the source code to build a model.

...