Anton Vlasjuk 9112520b15 Fix seq2seq collator padding (#30556) ...

* fix seq2seq data collator to respect the given padding strategy

further added tests for the seq2seq data collator in the style of the `data_collator_for_token_classification` (pt, tf, np)

* formatting and change bool equals "==" to "is"

* add missed return types in tests

* update numpy test as it can handle unequal shapes, not like pt or tf

2024-04-30 18:32:30 +01:00

..

__init__.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_data_collator.py

Fix seq2seq collator padding (#30556)

2024-04-30 18:32:30 +01:00

test_trainer_callback.py

Introduce Stateful Callbacks (#29666)

2024-04-25 11:00:09 -04:00

test_trainer_distributed.py

🚨 Fully revert atomic checkpointing 🚨 (#29370)

2024-03-04 06:17:42 -05:00

test_trainer_seq2seq.py

🚨🚨🚨Deprecate evaluation_strategy to eval_strategy🚨🚨🚨 (#30190)

2024-04-18 12:49:43 -04:00

test_trainer_tpu.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_trainer_utils.py

Add strategy to store results in evaluation loop (#30267)

2024-04-17 12:42:27 +01:00

test_trainer.py

Update unwrap from accelerate (#29933)

2024-04-19 18:05:34 +02:00