removed print in console
This commit is contained in:
parent
b910eeb4d0
commit
d101df2f57
@ -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)
|
Loading…
Reference in New Issue
Block a user