From 962cf868c960125170802294c79338adec391ffa Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sun, 10 Apr 2022 15:31:46 +0800 Subject: [PATCH] Fix import --- egs/librispeech/ASR/pruned_transducer_stateless2/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py index ddd2e8fb7..62dc825b6 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py @@ -45,6 +45,7 @@ from typing import Any, Dict, Optional, Tuple, Union import k2 import sentencepiece as spm import torch +import optim # from . import torch.multiprocessing as mp import torch.nn as nn from asr_datamodule import LibriSpeechAsrDataModule