rename ckpt suffix to np

4 years ago · 45a0e8c436
parent 1721626246
commit 45a0e8c436
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)