Add GigaAM NeMo transducer model for Russian ASR (#1467)
This commit is contained in:
@@ -75,6 +75,7 @@ def add_meta_data(filename: str, meta_data: Dict[str, str]):
|
||||
onnx.save(model, filename)
|
||||
|
||||
|
||||
@torch.no_grad()
|
||||
def main():
|
||||
model = EncDecCTCModel.from_config_file("./ctc_model_config.yaml")
|
||||
ckpt = torch.load("./ctc_model_weights.ckpt", map_location="cpu")
|
||||
|
||||
Reference in New Issue
Block a user