From d101df2f5701fff52ff1375b5dcaa950413eb078 Mon Sep 17 00:00:00 2001 From: Setra Solofoniaina <60129070+Setra-Solofoniaina@users.noreply.github.com> Date: Fri, 2 Apr 2021 15:49:58 +0300 Subject: [PATCH] removed print in console --- src/model/embedding/bert.py | 1 - 1 file changed, 1 deletion(-) 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