mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
e31c505c6c
commit
74ad69aa8d
Binary file not shown.
@ -401,6 +401,7 @@ class TransfEncoder(nn.TransformerEncoder):
|
|||||||
__constants__ = ['norm']
|
__constants__ = ['norm']
|
||||||
|
|
||||||
def __init__(self, encoder_layer, num_layers, norm=None, enable_nested_tensor=True, mask_check=True):
|
def __init__(self, encoder_layer, num_layers, norm=None, enable_nested_tensor=True, mask_check=True):
|
||||||
|
print(encoder_layer)
|
||||||
super(TransfEncoder, self).__init__(
|
super(TransfEncoder, self).__init__(
|
||||||
encoder_layer=encoder_layer,
|
encoder_layer=encoder_layer,
|
||||||
num_layers=num_layers,
|
num_layers=num_layers,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user