diff --git a/src/model/embedding/bert.py b/src/model/embedding/bert.py index 8d13edd..a5ec8a1 100644 --- a/src/model/embedding/bert.py +++ b/src/model/embedding/bert.py @@ -30,5 +30,4 @@ class BERTEmbedding(nn.Module): def forward(self, sequence, segment_label): x = self.token(sequence) + self.position(sequence) + self.segment(segment_label) - print(x.shape) return self.dropout(x) \ No newline at end of file