--- license: apache-2.0 base_model: google/vit-base-patch16-384 tags: - generated_from_keras_callback model-index: - name: Prahas10/roof-large results: [] --- # Prahas10/roof-large This model is a fine-tuned version of [google/vit-base-patch16-384](https://huggingface.co/google/vit-base-patch16-384) on an unknown dataset. It achieves the following results on the evaluation set: - Train Loss: 0.0498 - Validation Loss: 0.1617 - Train Accuracy: 0.9636 - Epoch: 9 ## 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: - optimizer: {'name': 'AdamWeightDecay', 'learning_rate': {'module': 'keras.optimizers.schedules', 'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 4e-05, 'decay_steps': 4380, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, 'registered_name': None}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.0001} - training_precision: float32 ### Training results | Train Loss | Validation Loss | Train Accuracy | Epoch | |:----------:|:---------------:|:--------------:|:-----:| | 1.6423 | 1.0616 | 0.7091 | 0 | | 0.4478 | 0.6131 | 0.8636 | 1 | | 0.1556 | 0.2951 | 0.9273 | 2 | | 0.0688 | 0.5952 | 0.7909 | 3 | | 0.1217 | 0.2658 | 0.9273 | 4 | | 0.0391 | 0.2163 | 0.9545 | 5 | | 0.0364 | 0.0717 | 0.9818 | 6 | | 0.0562 | 0.3374 | 0.8727 | 7 | | 0.0692 | 0.6530 | 0.7909 | 8 | | 0.0498 | 0.1617 | 0.9636 | 9 | ### Framework versions - Transformers 4.38.2 - TensorFlow 2.15.0 - Datasets 2.16.1 - Tokenizers 0.15.2