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: 172 Bytes
847d260 | 1 2 3 4 5 6 7 | {
"epoch": 800.0,
"train_loss": 0.36668021675006923,
"train_runtime": 517141.5399,
"train_samples_per_second": 44.339,
"train_steps_per_second": 5.543
} |