Minor fix

yzhangcs · yzhangcs · commit d41fd9979358 · 2021-08-02T14:59:38.000+08:00
diff --git a/supar/modules/transformer.py b/supar/modules/transformer.py
@@ -93,6 +93,7 @@ def __init__(self, n_layers, n_heads=8, n_model=1024, n_inner=2048, dropout=0.1)
         self.n_model = n_model
         self.n_inner = n_inner
 
+        self.pos_embed = SinusoidPositionalEmbedding()
         self.layers = nn.ModuleList([TransformerEncoderLayer(d_model=n_model,
                                                              nhead=n_heads,
                                                              dim_feedforward=n_inner,
