Commit 00ca52c6 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

add loggings

parent e203ae64
......@@ -85,7 +85,7 @@ def init_clustering(init_diar, cep, model_cfg):
"""
# Bic_lin is only applied when starting from a VAD segmentation or from UEM
if 'init_mode' in model_cfg['first_seg'] and not model_cfg['first_seg']['init_mode'] == "reference":
if model_cfg["model"]["vad"]["type"] != "reference":
output_diar = bic_linear_segmentation(init_diar, cep, model_cfg)
else:
output_diar = init_diar
......@@ -95,7 +95,7 @@ def init_clustering(init_diar, cep, model_cfg):
output_diar = cluster.perform()
# Viterbi devoding is only applied when starting from a VAD segmentation or from scratch
if 'init_mode' in model_cfg['first_seg'] and not model_cfg['first_seg']['init_mode'] == "reference":
if model_cfg["model"]["vad"]["type"] != "reference":
output_diar = s4d.viterbi.viterbi_decoding(cep, output_diar, model_cfg['first_seg']['thr_vit'])
return output_diar
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment