hnluo
2023-10-25 8a84ed6a4a6ac6958a480961dc2f1606b1ac7fb6
funasr/bin/diar_inference_launch.py
@@ -453,11 +453,17 @@
        help="The batch size for inference",
    )
    group.add_argument(
        "--diar_smooth_size",
        "--smooth_size",
        type=int,
        default=121,
        help="The smoothing size for post-processing"
    )
    group.add_argument(
        "--dur_threshold",
        type=int,
        default=10,
        help="The threshold of minimum duration"
    )
    return parser