File tree 1 file changed +10
-2
lines changed
1 file changed +10
-2
lines changed Original file line number Diff line number Diff line change @@ -453,10 +453,18 @@ index fc9411e95..21f1d3a43 100644
453
453
454
454
# Evaluate
455
455
diff --git a/examples/pytorch/image-classification/run_image_classification.py b/examples/pytorch/image-classification/run_image_classification.py
456
- index e2ce9f1c5..2a7fc1201 100644
456
+ index e2ce9f1c5..e0666475c 100644
457
457
--- a/examples/pytorch/image-classification/run_image_classification.py
458
458
+++ b/examples/pytorch/image-classification/run_image_classification.py
459
- @@ -275,6 +275,7 @@ def main():
459
+ @@ -230,6 +230,7 @@ def main():
460
+ cache_dir=model_args.cache_dir,
461
+ task="image-classification",
462
+ use_auth_token=True if model_args.use_auth_token else None,
463
+ + revision="014711311cec8b5959350c373878a3311caeb764",
464
+ )
465
+ else:
466
+ data_files = {}
467
+ @@ -275,6 +276,7 @@ def main():
460
468
id2label=id2label,
461
469
finetuning_task="image-classification",
462
470
cache_dir=model_args.cache_dir,
You can’t perform that action at this time.
0 commit comments