Fix train_step, test_step and tests for CLIP #18684
Rocketknight1
merged
32 commits into
main
from
return_loss_fix
gante
approved these changes
on 2022-08-18
sgugger
approved these changes
on 2022-08-31
Fix train_step and test_step, correctly enable CLIP fit test
4a3db4bb
Stop using get_args on older Python versions
2af8bba9
Don't use get_origin either
1107ead3
UnionType is actually even newer, don't use that either
1a331b49
Apply the same fix to test_loss_computation
0afc9fa6
Just realized I was accidentally skipping a bunch of tests!
e90bfcde
Fix test_loss_computation for models without separable labels
bb9f3e68
Fix scalar losses in test_step and train_step
48e28899
Stop committing your breakpoints
ca0dce9c
Fix Swin loss shape
2192f81b
Fix Tapas loss shape
046560e5
Shape fixes for TAPAS, DeIT, HuBERT and ViTMAE
3459ac09
Add loss computation to TFMobileBertForPreTraining
52b9baf1
make fixup and move copied from statement
6c3953bb
make fixup and move copied from statement
27c5616d
Correct copied from
13c7d9c5
Add labels and next_sentence_label inputs to TFMobileBERT
b76bed70
Make sure total_loss is always defined
01459276
Update tests/test_modeling_tf_common.py
15b96596
Fix copied from
b3a8ce41
Ensure CTC models get labels in tests
317824d1
Ensure CTC models get labels in tests
24382d78
Fix tests for vit_mae
518ecae3
Fix tests for vit_mae
c2170487
Fix tests for vit_mae
f4c47937
Reduce batch size for wav2vec2 testing because it was causing OOM
bf10b3bf
Skip some TAPAS tests that are failing
eea79ad3
Skip a failing HuBERT test
7804a90b
make style
47b2fe61
Fix mobilebertforpretraining test
406c7fdc
Skip Wav2Vec2 tests that use huge amounts of mem
4db13b9d
Skip keras_fit for Wav2Vec2 as well
641a20ba
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub