From 72cd3d474bd4d95ac31e27c9eaa8d0b9a87eccec Mon Sep 17 00:00:00 2001 From: YYJ-aaaa <1058387004@qq.com> Date: Wed, 6 Nov 2024 10:05:31 +0800 Subject: [PATCH] first commit --- .gitignore | 1 + config.json | 32 ++++++++++++++++++++++++++++++++ model.safetensors | 3 +++ preprocessor_config.json | 22 ++++++++++++++++++++++ pytorch_model.bin | 3 +++ training_args.bin | 3 +++ 6 files changed, 64 insertions(+) create mode 100644 .gitignore create mode 100644 config.json create mode 100644 model.safetensors create mode 100644 preprocessor_config.json create mode 100644 pytorch_model.bin create mode 100644 training_args.bin diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..0348ea9 --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +checkpoint-*/ \ No newline at end of file diff --git a/config.json b/config.json new file mode 100644 index 0000000..13fc6ea --- /dev/null +++ b/config.json @@ -0,0 +1,32 @@ +{ + "_name_or_path": "google/vit-base-patch16-224-in21k", + "architectures": [ + "ViTForImageClassification" + ], + "attention_probs_dropout_prob": 0.0, + "encoder_stride": 16, + "hidden_act": "gelu", + "hidden_dropout_prob": 0.0, + "hidden_size": 768, + "id2label": { + "0": "NORMAL", + "1": "PNEUMONIA" + }, + "image_size": 224, + "initializer_range": 0.02, + "intermediate_size": 3072, + "label2id": { + "NORMAL": 0, + "PNEUMONIA": 1 + }, + "layer_norm_eps": 1e-12, + "model_type": "vit", + "num_attention_heads": 12, + "num_channels": 3, + "num_hidden_layers": 12, + "patch_size": 16, + "problem_type": "single_label_classification", + "qkv_bias": true, + "torch_dtype": "float32", + "transformers_version": "4.30.2" +} diff --git a/model.safetensors b/model.safetensors new file mode 100644 index 0000000..cb1b059 --- /dev/null +++ b/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:505ad901b95bc3eb9c11e97715c1203b26c3e552c274b625ed7d968814d2efed +size 343223968 diff --git a/preprocessor_config.json b/preprocessor_config.json new file mode 100644 index 0000000..273d249 --- /dev/null +++ b/preprocessor_config.json @@ -0,0 +1,22 @@ +{ + "do_normalize": true, + "do_rescale": true, + "do_resize": true, + "image_mean": [ + 0.5, + 0.5, + 0.5 + ], + "image_processor_type": "ViTImageProcessor", + "image_std": [ + 0.5, + 0.5, + 0.5 + ], + "resample": 2, + "rescale_factor": 0.00392156862745098, + "size": { + "height": 224, + "width": 224 + } +} diff --git a/pytorch_model.bin b/pytorch_model.bin new file mode 100644 index 0000000..5f54d72 --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7d88f12de465387c770e7eb8a26ed77849b7e7b3cb3e374111266f3ce66a7dbf +size 343266993 diff --git a/training_args.bin b/training_args.bin new file mode 100644 index 0000000..91b839d --- /dev/null +++ b/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:da52dd1f4e287371d4584323b211c4752c866798fbed3cbdf4797c5cdc850bb3 +size 3951