def set_cfg_params_from_state(state: dict, cfg: DictConfig):
"""
Overrides some of the encoder config parameters from a give state object
"""
if not state:
return
cfg.do_lower_case = state["do_lower_case"]
if "encoder" in state:
saved_encoder_params = state["encoder"]
# TODO: try to understand why cfg.encoder = state["encoder"] doesn't work
for k, v in saved_encoder_params.items():
# TODO: tmp fix
if k == "q_wav2vec_model_cfg":
k = "q_encoder_model_cfg"
if k == "q_wav2vec_cp_file":
k = "q_encoder_cp_file"
if k == "q_wav2vec_cp_file":
k = "q_encoder_cp_file"
setattr(cfg.encoder, k, v)
else: # 'old' checkpoints backward compatibility support
pass
def set_cfg_params_from_state(state: dict, cfg: DictConfig):
"""
Overrides some of the encoder config parameters from a give state object
"""
if not state:
return
cfg.do_lower_case = state["do_lower_case"]
cfg.encoder.pretrained_model_cfg = state["pretrained_model_cfg"]
cfg.encoder.encoder_model_type = state["encoder_model_type"]
cfg.encoder.pretrained_file = state["pretrained_file"]
cfg.encoder.projection_dim = state["projection_dim"]
cfg.encoder.sequence_length = state["sequence_length"]
def set_cfg_params_from_state(state: dict, cfg: DictConfig):
"""
Overrides some of the encoder config parameters from a give state object
"""
if not state:
return
cfg.do_lower_case = state["do_lower_case"]
cfg.encoder.pretrained_model_cfg = state["pretrained_model_cfg"]
cfg.encoder.pretrained_file = state["pretrained_file"]
cfg.encoder.projection_dim = state["projection_dim"]
cfg.encoder.sequence_length = state["sequence_length"]
if cfg.encoder.encoder_model_type != state["encoder_model_type"]:
logger.warn(
f"Model type specified in the config file ('{cfg.encoder.encoder_model_type}') and model type "
f"obtained from the checkpoint ('{state['encoder_model_type']}') are different"
)
if not cfg.ignore_pretrained_model_type:
cfg.encoder.encoder_model_type = state["encoder_model_type"]
