fixed import bug.

Signed-off-by: Vahid <vnoroozi@nvidia.com>
NVIDIA · Nov 3, 2020 · 5e974b8 · 5e974b8
1 parent 353e652
commit 5e974b8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nemo/collections/asr/modules/conformer_encoder.py b/nemo/collections/asr/modules/conformer_encoder.py
@@ -14,7 +14,7 @@
 
 import math
 from collections import OrderedDict
-from collections.asr.parts.conformer_modules import ConformerEncoderBlock
+from nemo.collections.asr.parts.conformer_modules import ConformerEncoderBlock
 
 import torch
 import torch.nn as nn

diff --git a/nemo/collections/asr/parts/conformer_modules.py b/nemo/collections/asr/parts/conformer_modules.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from collections.asr.parts.multi_head_attention import MultiHeadAttention, RelPositionMultiHeadAttention
+from nemo.collections.asr.parts.multi_head_attention import MultiHeadAttention, RelPositionMultiHeadAttention
 
 import torch
 from torch import nn as nn