ViT fire-detection fine-tune
Hub data records image-classification with ViTForImageClassification and model type vit, and the model card describes a fine-tune of google/vit-base-patch16-224-in21k.
Open Source Model Profile · EdBianchi
vit-fire-detection is a ViT image-classification fine-tune from EdBianchi. According to the model card, it fine-tunes google/vit-base-patch16-224-in21k with reported 0.996 precision and recall.
vit-fire-detection is published by EdBianchi as a ViT image-classification model. The captured configuration identifies ViTForImageClassification with model type vit, and card data records apache-2.0. According to the model card, it is a fine-tune of google/vit-base-patch16-224-in21k.
Hub data records image-classification with ViTForImageClassification and model type vit, and the model card describes a fine-tune of google/vit-base-patch16-224-in21k.
According to the model card, the evaluation set reports loss 0.0126 with precision 0.9960 and recall 0.9960.
According to the model card, training used learning rate 0.0002, batch size 32, Adam, linear scheduling with 100 warmup steps, and 10 epochs.
Card data records apache-2.0 for this model.
Source: EdBianchi/vit-fire-detection
Captured: Unknown. Processed: 2026-09-07T19:34:30.609994+00:00.
vit-fire-detection This model is a fine-tuned version of google/vit-base-patch16-224-in21k on the None dataset. It achieves the following results on the evaluation set: Loss: 0.0126 Precision: 0.9960 Recall: 0.9960 Model description More information needed Intended uses & limitations More information needed Training and evaluation data More information needed Training procedure Training hyperparameters The following hyperparameters were used during training: learning_rate: 0.0002 train_batch_size: 32 eval_batch_size: 32 seed: 42 optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08 lr_scheduler_type: linear lr_scheduler_warmup_st…
F001F002F003F004F005F006F008F009F010F011F012