Hugging Face
Models
Datasets
Spaces
Posts
Docs
Solutions
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
6 months ago
.gitattributes
1.56 kB
Upload 12 files
6 months ago
README.md
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.
6 months ago
all_results.json
391 Bytes
Upload 12 files
6 months ago
config.json
1.1 kB
Upload 12 files
6 months ago
eval_results.json
203 Bytes
Upload 12 files
6 months ago
model.safetensors
350 MB
LFS
Upload 12 files
6 months ago
optimizer.pt
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
6 months ago
preprocessor_config.json
578 Bytes
Upload 12 files
6 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
6 months ago
scheduler.pt
pickle
Pickle imports
No problematic imports detected
What is a pickle import?
1.06 kB
LFS
Upload 12 files
6 months ago
state.db
350 MB
LFS
Upload 12 files
6 months ago
train_results.json
209 Bytes
Upload 12 files
6 months ago
trainer_state.json
44.4 kB
Upload 12 files
6 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
6 months ago
