Genre Classification — EfficientNet-B3
Music genre classifier trained on the Messy Mashup competition dataset.
Classes
blues, classical, country, disco, hiphop, jazz, metal, pop, reggae, rock
Architecture
EfficientNet-B3 fine-tuned on mel spectrograms (128 mel bins, 5s clips, 22050 Hz).
How to load
import torch
from torchvision import models
from huggingface_hub import hf_hub_download
model_path = hf_hub_download(repo_id="Kirandwip/Genre_prediction", filename="perfection_model_99.pth")
model = models.efficientnet_b3(weights=None)
model.classifier[1] = torch.nn.Linear(model.classifier[1].in_features, 10)
model.load_state_dict(torch.load(model_path, map_location="cpu"))
model.eval()