@@ -144,7 +144,7 @@ class FlaxDataCollatorForWav2Vec2Pretraining:
|
|||||||
The Wav2Vec2 model used for pretraining. The data collator needs to have access
|
The Wav2Vec2 model used for pretraining. The data collator needs to have access
|
||||||
to config and ``_get_feat_extract_output_lengths`` function for correct padding.
|
to config and ``_get_feat_extract_output_lengths`` function for correct padding.
|
||||||
feature_extractor (:class:`~transformers.Wav2Vec2FeatureExtractor`):
|
feature_extractor (:class:`~transformers.Wav2Vec2FeatureExtractor`):
|
||||||
The processor used for proccessing the data.
|
The processor used for processing the data.
|
||||||
padding (:obj:`bool`, :obj:`str` or :class:`~transformers.tokenization_utils_base.PaddingStrategy`, `optional`, defaults to :obj:`True`):
|
padding (:obj:`bool`, :obj:`str` or :class:`~transformers.tokenization_utils_base.PaddingStrategy`, `optional`, defaults to :obj:`True`):
|
||||||
Select a strategy to pad the returned sequences (according to the model's padding side and padding index)
|
Select a strategy to pad the returned sequences (according to the model's padding side and padding index)
|
||||||
among:
|
among:
|
||||||
|
|||||||
@@ -410,5 +410,5 @@ def generic_train(
|
|||||||
trainer.fit(model)
|
trainer.fit(model)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
print("RAG modeling tests with new set functions successfuly executed!")
|
print("RAG modeling tests with new set functions successfully executed!")
|
||||||
return trainer
|
return trainer
|
||||||
|
|||||||
Reference in New Issue
Block a user