How to use gary2002/output_dir with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="gary2002/output_dir") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")
# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("gary2002/output_dir") model = AutoModelForImageClassification.from_pretrained("gary2002/output_dir", device_map="auto")
4055d8a 47b7b12 4055d8a
1
2
3
4
5
6
7
8
{ "epoch": 3.0, "total_flos": 1.6948561394167603e+17, "train_loss": 0.7343477571391623, "train_runtime": 162.6903, "train_samples_per_second": 13.443, "train_steps_per_second": 1.346 }
