config.json

{"module": "keras_hub.src.models.vit.vit_image_classifier", "class_name": "ViTImageClassifier", "config": {"backbone": {"module": "keras_hub.src.models.vit.vit_backbone", "class_name": "ViTBackbone", "config": {"name": "vi_t_backbone", "trainable": true, "image_shape": [224, 224, 3], "patch_size": 16, "num_layers": 12, "num_heads": 12, "hidden_dim": 768, "mlp_dim": 3072, "dropout_rate": 0.0, "attention_dropout": 0.0, "layer_norm_epsilon": 1e-06, "use_mha_bias": true, "use_mlp_bias": true}, "registered_name": "keras_hub>ViTBackbone", "build_config": {"input_shape": null}}, "preprocessor": {"module": "keras_hub.src.models.vit.vit_image_classifier_preprocessor", "class_name": "ViTImageClassifierPreprocessor", "config": {"name": "vi_t_image_classifier_preprocessor", "trainable": true, "dtype": {"module": "keras", "class_name": "DTypePolicy", "config": {"name": "float32"}, "registered_name": null}, "image_converter": {"module": "keras_hub.src.models.vit.vit_image_converter", "class_name": "ViTImageConverter", "config": {"name": "vi_t_image_converter", "trainable": true, "dtype": {"module": "keras", "class_name": "DTypePolicy", "config": {"name": "float32"}, "registered_name": null}, "image_size": [224, 224], "scale": 0.00392156862745098, "offset": null, "interpolation": "bilinear", "antialias": false, "crop_to_aspect_ratio": true, "pad_to_aspect_ratio": false, "bounding_box_format": "yxyx", "norm_mean": [0.5, 0.5, 0.5], "norm_std": [0.5, 0.5, 0.5]}, "registered_name": "keras_hub>ViTImageConverter"}, "config_file": "preprocessor.json"}, "registered_name": "keras_hub>ViTImageClassifierPreprocessor"}, "name": "vi_t_image_classifier", "num_classes": 6, "pooling": "token", "activation": "softmax", "dropout": 0.0, "intermediate_dim": null}, "registered_name": "keras_hub>ViTImageClassifier", "build_config": {"input_shape": null}, "compile_config": {"optimizer": {"module": "keras.optimizers", "class_name": "Adam", "config": {"name": "adam", "learning_rate": 4.999999873689376e-05, "weight_decay": null, "clipnorm": null, "global_clipnorm": null, "clipvalue": null, "use_ema": false, "ema_momentum": 0.99, "ema_overwrite_frequency": null, "loss_scale_factor": null, "gradient_accumulation_steps": null, "beta_1": 0.9, "beta_2": 0.999, "epsilon": 1e-07, "amsgrad": false}, "registered_name": null}, "loss": {"module": "keras.losses", "class_name": "SparseCategoricalCrossentropy", "config": {"name": "sparse_categorical_crossentropy", "reduction": "sum_over_batch_size", "from_logits": true, "ignore_class": null}, "registered_name": null}, "loss_weights": null, "metrics": [{"module": "keras.metrics", "class_name": "SparseCategoricalAccuracy", "config": {"name": "sparse_categorical_accuracy", "dtype": "float32"}, "registered_name": null}], "weighted_metrics": null, "run_eagerly": false, "steps_per_execution": 1, "jit_compile": true}}