| | |
| | | # 2. Build Language model |
| | | if lm_train_config is not None: |
| | | lm, lm_train_args = build_model_from_file( |
| | | lm_train_config, lm_file, device |
| | | lm_train_config, lm_file, None, device, task_name="lm" |
| | | ) |
| | | scorers["lm"] = lm.lm |
| | | |
| | |
| | | # 2. Build Language model |
| | | if lm_train_config is not None: |
| | | lm, lm_train_args = build_model_from_file( |
| | | lm_train_config, lm_file, device |
| | | lm_train_config, lm_file, None, device, task_name="lm" |
| | | ) |
| | | scorers["lm"] = lm.lm |
| | | |
| | |
| | | # 2. Build Language model |
| | | if lm_train_config is not None: |
| | | lm, lm_train_args = build_model_from_file( |
| | | lm_train_config, lm_file, device |
| | | lm_train_config, lm_file, None, device, task_name="lm" |
| | | ) |
| | | lm.to(device) |
| | | scorers["lm"] = lm.lm |
| | |
| | | |
| | | if lm_train_config is not None: |
| | | lm, lm_train_args = build_model_from_file( |
| | | lm_train_config, lm_file, device |
| | | lm_train_config, lm_file, None, device, task_name="lm" |
| | | ) |
| | | lm_scorer = lm.lm |
| | | else: |
| | |
| | | # 2. Build Language model |
| | | if lm_train_config is not None: |
| | | lm, lm_train_args = build_model_from_file( |
| | | lm_train_config, lm_file, None, device |
| | | lm_train_config, lm_file, None, device, task_name="lm" |
| | | ) |
| | | scorers["lm"] = lm.lm |
| | | |