ref: f3b86f941408b37b0c0236eb5b8b09605b8a713b
parent: 0459a572f592fb07376c480c1ebbf04c16090211
author: Jean-Marc Valin <jmvalin@amazon.com>
date: Fri Sep 29 22:48:26 EDT 2023
Fix model saving
--- a/dnn/torch/neural-pitch/training.py
+++ b/dnn/torch/neural-pitch/training.py
@@ -148,7 +148,7 @@
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
--
⑨