diff --git a/dnn/torch/neural-pitch/training.py b/dnn/torch/neural-pitch/training.py
index 04b3deb1b85d7a7150c7d2d3ad336803e3f260d8..e725e57c631722a44adc9c178689dfa904d6da63 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