Skip to content

Commit

Permalink
fix: fix training_stride eds.transformer parameter
Browse files Browse the repository at this point in the history
percevalw committed May 20, 2024
1 parent 9cb78aa commit 3ea89f0
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion edsnlp/pipes/trainable/embeddings/transformer/transformer.py
Original file line number Diff line number Diff line change
@@ -288,7 +288,9 @@ def collate(self, batch):
-------
"""
stride = self.window if self.training and self.training_stride else self.stride
stride = (
self.window if self.training and not self.training_stride else self.stride
)
max_seq_size = max(
[
2 # CLS and SEP tokens

0 comments on commit 3ea89f0

Please sign in to comment.