diff --git a/config.json b/config.json new file mode 100644 index 0000000..ec7eaa0 --- /dev/null +++ b/config.json @@ -0,0 +1,50 @@ +{ + "_name_or_path": "AutoTrain", + "architectures": [ + "SwinForImageClassification" + ], + "attention_probs_dropout_prob": 0.0, + "depths": [ + 2, + 2, + 18, + 2 + ], + "drop_path_rate": 0.1, + "embed_dim": 128, + "encoder_stride": 32, + "hidden_act": "gelu", + "hidden_dropout_prob": 0.0, + "hidden_size": 1024, + "id2label": { + "0": "artificial", + "1": "human" + }, + "image_size": 224, + "initializer_range": 0.02, + "label2id": { + "artificial": "0", + "human": "1" + }, + "layer_norm_eps": 1e-05, + "max_length": 128, + "mlp_ratio": 4.0, + "model_type": "swin", + "num_channels": 3, + "num_heads": [ + 4, + 8, + 16, + 32 + ], + "num_layers": 4, + "padding": "max_length", + "patch_size": 4, + "path_norm": true, + "problem_type": "single_label_classification", + "qkv_bias": true, + "torch_dtype": "float32", + "transformers_version": "4.22.1", + "use_absolute_embeddings": false, + "window_size": 7 +} diff --git a/preprocessor_config.json b/preprocessor_config.json new file mode 100644 index 0000000..45e3931 --- /dev/null +++ b/preprocessor_config.json @@ -0,0 +1,17 @@ +{ + "do_normalize": true, + "do_resize": true, + "feature_extractor_type": "ViTFeatureExtractor", + "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..4d8c673 --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f4c14ed23eb1d65b1c6ca7b163f8f91bebcf3a4d8c275296c850c9f1bbd01daf +size 347596479