Add magic method to our TF models to convert datasets with column inference #17160
Rocketknight1
marked this pull request as ready for review 4 years ago
gante
commented
on 2022-05-19
Add method to call to_tf_dataset() with column inference
1ccd7260
Add test for dataset creation
2c71f846
Add a default arg for data collator
919cf82d
Fix test
0066598e
Fix call with non-dev version of datasets
b40fa6eb
Test correct column removal too
f5f667d0
make fixup
258392be
More tests to make sure we remove unwanted columns
ae4be4af
Fix test to avoid predicting on unbuilt models
673e23d6
Fix test to avoid predicting on unbuilt models
0ee6e1da
Fix test to remove unwanted head mask columns from inputs
2313b3ad
Stop pushing your debug breakpoints to the main repo of the $2bn comp…
221ae783
Skip the test in convnext because no grouped conv support
15061825
Drop bools from the dataset dict
a9010b10
Make style
0a29747e
Skip the training test for models whose input dicts don't give us labels
5b35ff4a
Skip transformerXL in the test because it doesn't return a simple loss
a1b6e924
Skip TFTapas because of some odd NaN losses
33812eaa
make style
fb19ea9f
make fixup
a7f6a859
Add docstring
0787a458
fixup
24c0a66c
Update src/transformers/modeling_tf_utils.py
fc89e11d
Update src/transformers/modeling_tf_utils.py
0d4553dd
Update src/transformers/modeling_tf_utils.py
03111729
Update src/transformers/modeling_tf_utils.py
a7b1f601
Update src/transformers/modeling_tf_utils.py
f01f8c21
Remove breakpoint from tests
24cc6a6c
Fix assert, add requires_backends
e381daff
Protect tokenizer import with if TYPE_CHECKING
0f473db8
make fixup
753c0c50
Add noqa, more fixup
b23efa1e
More rearranging for ~* aesthetics *~
163b4f72
Adding defaults for shuffle and batch_size to match to_tf_dataset()
e29f98fe
Update src/transformers/modeling_tf_utils.py
726cb39b
gante
approved these changes
on 2022-05-19
sgugger
approved these changes
on 2022-05-19
Rocketknight1
deleted the magic_tf_dataset_method branch 4 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub