diff --git a/modyn/models/tokenizers/distill_bert_tokenizer.py b/modyn/models/tokenizers/distill_bert_tokenizer.py index 37f0f8fb8..7b9e44a31 100644 --- a/modyn/models/tokenizers/distill_bert_tokenizer.py +++ b/modyn/models/tokenizers/distill_bert_tokenizer.py @@ -8,6 +8,7 @@ class DistilBertTokenizerTransform(HFTokenizerTransform): def __init__(self, max_token_length: int = 300): """Transformer for DistilBERT. + Args: max_token_length: Maximum token length for DistilBERT. """