diff options
| author | Jean-Marc Valin <jmvalin@amazon.com> | 2023-09-30 09:48:26 +0300 |
|---|---|---|
| committer | Jean-Marc Valin <jmvalin@amazon.com> | 2023-09-30 09:48:26 +0300 |
| commit | f3b86f941408b37b0c0236eb5b8b09605b8a713b (patch) | |
| tree | 554a73cafd9ff24362586d0b64da134a5b8f5693 | |
| parent | 0459a572f592fb07376c480c1ebbf04c16090211 (diff) | |
Fix model saving
| -rw-r--r-- | dnn/torch/neural-pitch/training.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/dnn/torch/neural-pitch/training.py b/dnn/torch/neural-pitch/training.py index 04b3deb1..e725e57c 100644 --- a/dnn/torch/neural-pitch/training.py +++ b/dnn/torch/neural-pitch/training.py @@ -148,7 +148,7 @@ config = dict( context=args.context, ) -model_save_path = os.path.join(args.output, f"{args.prefix}_{args.data_format}.pth") +model_save_path = os.path.join(args.output_folder, f"{args.prefix}_{args.data_format}.pth") checkpoint = { 'state_dict': pitch_nn.state_dict(), 'config': config |