Removing training-only nodes

TensorFlow

The Arm NN SDK is an inference-only engine. For performance and compatibility, you must remove training-only nodes when using the TensorFlow neural network framework. These nodes include operations that you do not require for inference. 

TensorFlow provides documentation on how to remove these training-only nodes.

Caffe

You do not need to remove training-only nodes when you use the Caffe neural network framework as the Arm NN SDK handles the removal.

Previous Next