Update README.md
Browse files
README.md
CHANGED
|
@@ -70,7 +70,7 @@ feat = feat.squeeze(0).cpu().numpy()
|
|
| 70 |
print(f"Feature shape: {feat.shape}")
|
| 71 |
# end of feature extraction
|
| 72 |
|
| 73 |
-
## get top
|
| 74 |
with torch.no_grad():
|
| 75 |
pred = model(mel)
|
| 76 |
|
|
|
|
| 70 |
print(f"Feature shape: {feat.shape}")
|
| 71 |
# end of feature extraction
|
| 72 |
|
| 73 |
+
## get top 12 predictions
|
| 74 |
with torch.no_grad():
|
| 75 |
pred = model(mel)
|
| 76 |
|