mode

Parameters

mode

The computation mode to use (INFERENCE or TRAINING)