Commit ac21b5ec authored by peastman's avatar peastman
Browse files

Added missing default argument

parent 426ae2b7
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -770,7 +770,7 @@ class TensorGraph(Model):
    return tf.train.get_checkpoint_state(
        self.model_dir).all_model_checkpoint_paths

  def restore(self, checkpoint):
  def restore(self, checkpoint=None):
    """Reload the values of all variables from a checkpoint file.

    Parameters