xcx0902
/

vit_mnist

Image Classification

Model card Files Files and versions

Ctrl+K

Ctrl+K

1 contributor

History: 4 commits

xcx0902's picture

Create preprocessor_config.json

6fe2f9b verified about 1 month ago

.gitattributes

1.52 kB

initial commit about 1 month ago
README.md

82 Bytes

Create README.md about 1 month ago
config.json

1.03 kB

Upload folder using huggingface_hub about 1 month ago
model.safetensors

2.22 MB
xet

Upload folder using huggingface_hub about 1 month ago
preprocessor_config.json

216 Bytes

Create preprocessor_config.json about 1 month ago
train.py

3.35 kB

Upload folder using huggingface_hub about 1 month ago
vit_mnist.pth
Detected Pickle imports (24)
- "transformers.models.vit.modeling_vit.ViTLayer",
- "torch._utils._rebuild_tensor_v2",
- "transformers.models.vit.configuration_vit.ViTConfig",
- "torch.nn.modules.linear.Linear",
- "transformers.models.vit.modeling_vit.ViTIntermediate",
- "torch._C._nn.gelu",
- "torch.nn.modules.container.ModuleList",
- "transformers.models.vit.modeling_vit.ViTPatchEmbeddings",
- "transformers.models.vit.modeling_vit.ViTForImageClassification",
- "transformers.models.vit.modeling_vit.ViTSelfOutput",
- "torch.FloatStorage",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.conv.Conv2d",
- "transformers.models.vit.modeling_vit.ViTModel",
- "torch.nn.modules.dropout.Dropout",
- "transformers.models.vit.modeling_vit.ViTEmbeddings",
- "transformers.models.vit.modeling_vit.ViTEncoder",
- "transformers.models.vit.modeling_vit.ViTAttention",
- "transformers.activations.GELUActivation",
- "transformers.models.vit.modeling_vit.ViTOutput",
- "torch.nn.modules.normalization.LayerNorm",
- "collections.OrderedDict",
- "__builtin__.set",
- "transformers.models.vit.modeling_vit.ViTSelfAttention"
How to fix it?
2.25 MB
xet

Upload folder using huggingface_hub about 1 month ago