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

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
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.