removed print in console

This commit is contained in:
Setra Solofoniaina 2021-04-02 15:49:58 +03:00
parent b910eeb4d0
commit d101df2f57

View File

@ -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)