Spaces:

qgyd2021
/

nx_denoise

Running

HoneyTian commited on Mar 3

Commit

a2985be

1 Parent(s): 0f4cf3f

update

Files changed (2) hide show

toolbox/torchaudio/models/nx_clean_unet/transformer/attention.py CHANGED Viewed

@@ -7,7 +7,7 @@ import torch
 import torch.nn as nn
-class MultiHeadedAttention(nn.Module):
     def __init__(self, n_head: int, n_feat: int, dropout_rate: float):
         """
         :param n_head: int. the number of heads.

 import torch.nn as nn
+class MultiHeadAttention(nn.Module):
     def __init__(self, n_head: int, n_feat: int, dropout_rate: float):
         """
         :param n_head: int. the number of heads.

toolbox/torchaudio/models/nx_clean_unet/transformer/transformer.py CHANGED Viewed

@@ -7,7 +7,7 @@ import torch
 import torch.nn as nn
 from toolbox.torchaudio.models.nx_clean_unet.transformer.mask import subsequent_chunk_mask
-from toolbox.torchaudio.models.nx_clean_unet.transformer.attention import MultiHeadedAttention, RelativeMultiHeadedAttention
 class PositionwiseFeedForward(nn.Module):

 import torch.nn as nn
 from toolbox.torchaudio.models.nx_clean_unet.transformer.mask import subsequent_chunk_mask
+from toolbox.torchaudio.models.nx_clean_unet.transformer.attention import MultiHeadAttention, RelativeMultiHeadSelfAttention
 class PositionwiseFeedForward(nn.Module):