SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add TF contrastive image text finetuning example (#21939)

Browse Source 
* Initial commit

* stash commit

* Add model checkpointing and pushing

* Fix model name inference

* Update README

* Update README

* Remove a couple of Torch references

* Update copyright date

* make fixup

* Update PushToHubCallback args!

* Remove the torch summary

* Add strategy.scope
This commit is contained in:
Matt
2023-03-06 16:57:40 +00:00
committed by GitHub
parent 9474abdf47
commit 5d8efc79db
13 changed files with 686 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/tensorflow/text-classification/run_text_classification.py
View File

@@ -502,9 +502,8 @@ def main():
callbacks = [
PushToHubCallback(
output_dir=training_args.output_dir,
model_id=push_to_hub_model_id,
organization=training_args.push_to_hub_organization,
token=training_args.push_to_hub_token,
hub_model_id=push_to_hub_model_id,
hub_token=training_args.push_to_hub_token,
tokenizer=tokenizer,
**model_card_kwargs,
)