This commit is contained in:
语帆 2024-02-23 16:17:18 +08:00
parent 5c1308e3cf
commit 54bd357b08

View File

@ -355,7 +355,7 @@ class SelfSrcAttention(nn.Module):
return x, tgt_mask, memory, memory_mask
@tables.register("encoder_classes", "ConvBiasPredictor")
class ConvPredictor(nn.Module):
def __init__(self, size=256, l_order=3, r_order=3, attention_heads=4, attention_dropout_rate=0.1, linear_units=2048):
super().__init__()