Transformers
TensorBoard
Safetensors
vit_mae
pretraining
masked-auto-encoding
Generated from Trainer
Instructions to use jaypratap/vit-pretraining-2024_03_14 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use jaypratap/vit-pretraining-2024_03_14 with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, AutoModelForPreTraining processor = AutoImageProcessor.from_pretrained("jaypratap/vit-pretraining-2024_03_14") model = AutoModelForPreTraining.from_pretrained("jaypratap/vit-pretraining-2024_03_14", device_map="auto") - Notebooks
- Google Colab
- Kaggle
File size: 313 Bytes
847d260 | 1 2 3 4 5 6 7 8 9 10 11 | {
"epoch": 800.0,
"eval_loss": 0.3360130190849304,
"eval_runtime": 63.6007,
"eval_samples_per_second": 79.543,
"eval_steps_per_second": 9.953,
"train_loss": 0.36668021675006923,
"train_runtime": 517141.5399,
"train_samples_per_second": 44.339,
"train_steps_per_second": 5.543
} |