Method save

rat-sql-gap/seq2struct/utils/saver.py:125–133

Saves model and optimizer to given directory. Args: model_dir: Model directory to save. step: Current training step.

(self, model_dir, step)

        return last_step

    def save(self, model_dir, step):
        """Saves model and optimizer to given directory.

        Args:
           model_dir: Model directory to save.
           step: Current training step.
        """
        save_checkpoint(self._items, step, model_dir,
                        keep_every_n=self._keep_every_n)

    def restore_part(self, other_model_dir, remap):
        """Restores part of the model from other directory.