// TODO: This is not the best place to configure rank? Why is rank not
// transmitted through the queue
train_image.set_shape((None, None, 3))
// We add fake batch dimension to train data. TODO: DEFINITELY NOT THE BEST
// PLACE
train_image = tf.expand_dims(train_image, 0)
After Change
// If the config file is empty, our config will be the base_config for the
// default model.
try:
config = get_config(config_files, override_params=override_params)
except KeyError:
raise KeyError("model.type should be set on the custom config.")
config.train.job_dir = job_dir or config.train.job_dir