Hugging Face
Models
Datasets
Spaces
Posts
Docs
Enterprise
Pricing
Log In
Sign Up
KaushalB
/
ViTForMusicClassification
like
1
Image Classification
Transformers
Safetensors
ghermoso/egtzan_plus
vit
ViT
music
CV
Inference Endpoints
License:
apache-2.0
Model card
Files
Files and versions
Community
Train
Deploy
Use this model
main
ViTForMusicClassification
1 contributor
History:
3 commits
KaushalB
This is an implementation of the Google's Vision Transformer large patch 32 that is used for music classification into different genres. The dataset used is the gtzan dataset which has melspectrograms of many songs.
faba7b5
verified
7 months ago
.gitattributes
Safe
1.56 kB
Upload 12 files
8 months ago
README.md
Safe
167 Bytes
This is an implementation of the Google's Vision Transformer large patch 32 that is used for music classification into different genres. The dataset used is the gtzan dataset which has melspectrograms of many songs.
7 months ago
all_results.json
Safe
391 Bytes
Upload 12 files
8 months ago
config.json
Safe
1.1 kB
Upload 12 files
8 months ago
eval_results.json
Safe
203 Bytes
Upload 12 files
8 months ago
model.safetensors
Safe
350 MB
LFS
Upload 12 files
8 months ago
optimizer.pt
Safe
pickle
Detected Pickle imports (3)
"torch.FloatStorage"
,
"torch._utils._rebuild_tensor_v2"
,
"collections.OrderedDict"
What is a pickle import?
700 MB
LFS
Upload 12 files
8 months ago
preprocessor_config.json
Safe
578 Bytes
Upload 12 files
8 months ago
rng_state.pth
pickle
Detected Pickle imports (7)
"numpy.core.multiarray._reconstruct"
,
"torch._utils._rebuild_tensor_v2"
,
"torch.ByteStorage"
,
"collections.OrderedDict"
,
"numpy.ndarray"
,
"_codecs.encode"
,
"numpy.dtype"
How to fix it?
14.2 kB
LFS
Upload 12 files
8 months ago
scheduler.pt
Safe
pickle
Pickle imports
No problematic imports detected
What is a pickle import?
1.06 kB
LFS
Upload 12 files
8 months ago
state.db
Safe
350 MB
LFS
Upload 12 files
8 months ago
train_results.json
Safe
209 Bytes
Upload 12 files
8 months ago
trainer_state.json
Safe
44.4 kB
Upload 12 files
8 months ago
training_args.bin
pickle
Detected Pickle imports (9)
"transformers.trainer_utils.IntervalStrategy"
,
"transformers.training_args.OptimizerNames"
,
"transformers.trainer_pt_utils.AcceleratorConfig"
,
"accelerate.utils.dataclasses.DistributedType"
,
"transformers.trainer_utils.HubStrategy"
,
"accelerate.state.PartialState"
,
"transformers.trainer_utils.SchedulerType"
,
"torch.device"
,
"transformers.training_args.TrainingArguments"
How to fix it?
4.92 kB
LFS
Upload 12 files
8 months ago