diff --git a/config.json b/config.json new file mode 100644 index 0000000..b52f479 --- /dev/null +++ b/config.json @@ -0,0 +1,55 @@ +{ + "_name_or_path": "ahsanjavid/conVneXt-tiny-finetuned-cifar10", + "architectures": [ + "ConvNextForImageClassification" + ], + "depths": [ + 3, + 3, + 9, + 3 + ], + "drop_path_rate": 0.0, + "hidden_act": "gelu", + "hidden_sizes": [ + 96, + 192, + 384, + 768 + ], + "id2label": { + "0": "airplane", + "1": "automobile", + "2": "bird", + "3": "cat", + "4": "deer", + "5": "dog", + "6": "frog", + "7": "horse", + "8": "ship", + "9": "truck" + }, + "image_size": 224, + "initializer_range": 0.02, + "label2id": { + "airplane": 0, + "automobile": 1, + "bird": 2, + "cat": 3, + "deer": 4, + "dog": 5, + "frog": 6, + "horse": 7, + "ship": 8, + "truck": 9 + }, + "layer_norm_eps": 1e-12, + "layer_scale_init_value": 1e-06, + "model_type": "convnext", + "num_channels": 3, + "num_stages": 4, + "patch_size": 4, + "problem_type": "single_label_classification", + "torch_dtype": "float32", + "transformers_version": "4.21.1" +} diff --git a/preprocessor_config.json b/preprocessor_config.json new file mode 100644 index 0000000..9a46cca --- /dev/null +++ b/preprocessor_config.json @@ -0,0 +1,18 @@ +{ + "crop_pct": 0.875, + "do_normalize": true, + "do_resize": true, + "feature_extractor_type": "ConvNextFeatureExtractor", + "image_mean": [ + 0.485, + 0.456, + 0.406 + ], + "image_std": [ + 0.229, + 0.224, + 0.225 + ], + "resample": 3, + "size": 224 +} diff --git a/pytorch_model.bin b/pytorch_model.bin new file mode 100644 index 0000000..1e375d1 --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ef740d75263b2a7cd022e846f8f9ecb150af9398273c47a97081bd8af8129205 +size 111369365