metadata
library_name: pytorch
license: mit
pipeline_tag: image-classification
tags:
- vision-transformer
- walkability
ViT-Walkability
This repository contains a fine-tuned Vision Transformer model for walkability-related image classification tasks.
Checkpoint details
- Source file:
best.pt - Top-level type:
dict - Base model:
google/vit-base-patch16-224 - Num labels:
5 - Labels:
['LABEL_0', 'LABEL_1', 'LABEL_2', 'LABEL_3', 'LABEL_4'] - Keys:
['model_state_dict', 'epoch']
Loading notes
The uploaded files are in standard Hugging Face Transformers format and can be loaded directly with from_pretrained.