Fix checkpoint loader to use GFile
@@ -395,7 +395,7 @@ def train(config_file, experiment_id=None, recreate=False):
if not tf.io.gfile.exists(checkpoint):
raise Exception('Checkpoints are not created in {}'.format(experiment_dir))
- with open(checkpoint) as stream:
+ with tf.io.gfile.GFile(checkpoint) as stream:
data = yaml.load(stream)
checkpoint_name = os.path.basename(data['model_checkpoint_path'])
