Views
No views yet
google/vit-base-patch16-224 для классификации
изображений на три класса: cat, dog, panda.[CLS] токене. Стратегия дообучения: linear_probe.(B, 3, 224, 224).linear_probe_best.pt — чекпойнт PyTorch (torch.save) со словарём:
model_state, class_names, model_name, strategy, epoch, val_acc.1from huggingface_hub import hf_hub_download
2from src.config import Config
3from src.inference import Predictor
4
5path = hf_hub_download(repo_id="A11Sunday/vit-cat-dog-panda", filename="linear_probe_best.pt")
6predictor = Predictor(Config(), path)
7print(predictor.predict(image)) # {"cat": ..., "dog": ..., "panda": ...}