This is to allow for using the a checkpoint resulting from training (exported to a saved model) for predictions. This will form the basis for defining the high-level interface - instances in, predictions out.