When using TensorFlow, it is recommended that you wrap any instance of
This guarantees that only one checkpoint is saved per
tf.train.Saver().save(...) call, regardless of how many
GPUs or model replicas there are.
import engineml.tensorflow as eml import tensorflow as tf saver = tf.train.Saver() saver = eml.saver(saver)