rename ckpt suffix to np

3 years ago · a997b5a61c
parent 0f3e5a3872
commit a997b5a61c
1 changed files with 1 additions and 1 deletions
--- a/deepspeech/training/extensions/snapshot.py
+++ b/deepspeech/training/extensions/snapshot.py
@ -101,7 +101,7 @@ class Snapshot(extension.Extension):
        iteration = trainer.updater.state.iteration
        epoch = trainer.updater.state.epoch
        num = epoch if self.trigger[1] == 'epoch' else iteration
-        path = self.checkpoint_dir / f"{num}.pdz"
+        path = self.checkpoint_dir / f"{num}.np"

        # add the new one
        trainer.updater.save(path)