riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 18:22:34 +06:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

keep same dataset actually

Browse Source
This commit is contained in:
Quentin Lhoest 2025-06-26 17:05:55 +02:00 committed by ydshieh
parent 79c9aece59
commit d20ab64120
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/test_accelerate_examples.py
View File

@ -341,7 +341,7 @@ class ExamplesTestsNoTrainer(TestCasePlus):
testargs = f""" testargs = f"""
{self.examples_dir}/pytorch/object-detection/run_object_detection_no_trainer.py {self.examples_dir}/pytorch/object-detection/run_object_detection_no_trainer.py
--model_name_or_path qubvel-hf/detr-resnet-50-finetuned-10k-cppe5 --model_name_or_path qubvel-hf/detr-resnet-50-finetuned-10k-cppe5
--dataset_name hf-internal-testing/cppe-5-sample --dataset_name qubvel-hf/cppe-5-sample
--output_dir {tmp_dir} --output_dir {tmp_dir}
--max_train_steps=10 --max_train_steps=10
--num_warmup_steps=2 --num_warmup_steps=2

2
examples/pytorch/test_pytorch_examples.py
View File

@ -620,7 +620,7 @@ class ExamplesTests(TestCasePlus):
run_object_detection.py run_object_detection.py
--model_name_or_path qubvel-hf/detr-resnet-50-finetuned-10k-cppe5 --model_name_or_path qubvel-hf/detr-resnet-50-finetuned-10k-cppe5
--output_dir {tmp_dir} --output_dir {tmp_dir}
--dataset_name hf-internal-testing/cppe-5-sample --dataset_name qubvel-hf/cppe-5-sample
--do_train --do_train
--do_eval --do_eval
--remove_unused_columns False --remove_unused_columns False