from local

This commit is contained in:
dohe0342 2023-02-14 17:45:16 +09:00
parent 9bc97cd8c1
commit b2ab9a6bd2
2 changed files with 3 additions and 0 deletions

View File

@ -141,6 +141,9 @@ class Conformer(Transformer):
self.layer_norm = nn.LayerNorm(d_model)
self.interctc = interctc
self.interctc_condition = interctc_condition
if self.interctc_condition:
self.condition_layer = ScaledLinear(d_model, 500)
def run_encoder(
self,