|
@@ -57,6 +57,9 @@ parser.add_argument("--confusion-clips", choices=["recording", "clip"], default=
|
|
|
parser.add_argument("--confusion-groups", choices=["child", "recording"], default="recording")
|
|
|
parser.add_argument("--time-scale", type=int, default=0)
|
|
|
parser.add_argument("--require-clip-age", action="store_true", default=True)
|
|
|
+parser.add_argument("--adapt-delta", type=float, default=0.9)
|
|
|
+parser.add_argument("--max-treedepth", type=int, default=14)
|
|
|
+
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
speakers = ["CHI", "OCH", "FEM", "MAL"]
|
|
@@ -285,7 +288,8 @@ def run_model(data, run, model_name):
|
|
|
iter_sampling=args.samples,
|
|
|
iter_warmup=args.warmup,
|
|
|
step_size=0.1,
|
|
|
- max_treedepth=14,
|
|
|
+ max_treedepth=args.max_treedepth,
|
|
|
+ adapt_delta=args.adapt_delta,
|
|
|
# save_profile=True,
|
|
|
show_console=True,
|
|
|
)
|