Transformers
TensorBoard
Safetensors
vit_mae
pretraining
masked-auto-encoding
Generated from Trainer
Instructions to use jaypratap/vit-pretraining-2024_03_10 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use jaypratap/vit-pretraining-2024_03_10 with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, AutoModelForPreTraining processor = AutoImageProcessor.from_pretrained("jaypratap/vit-pretraining-2024_03_10") model = AutoModelForPreTraining.from_pretrained("jaypratap/vit-pretraining-2024_03_10", device_map="auto") - Notebooks
- Google Colab
- Kaggle
File size: 170 Bytes
b52facb | 1 2 3 4 5 6 7 | {
"epoch": 200.0,
"train_loss": 0.513909718911276,
"train_runtime": 133465.2589,
"train_samples_per_second": 29.286,
"train_steps_per_second": 3.661
} |