in tensorflow_managed_spot_training_checkpointing/generate_cifar10_tfrecords.py [0:0]
def read_pickle_from_file(filename):
with tf.io.gfile.GFile(filename, 'rb') as f:
if sys.version_info.major >= 3:
return pickle.load(f, encoding='bytes')
else:
return pickle.load(f)