joel/freqsplit
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

enable cuda for classify_audio and demucs_wrapper

Browse Source
This commit is contained in:
Joel Mathew Thomas
2025-01-25 17:38:27 +05:30
parent e0c19002d1
commit 7295d4ff3d
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/preprocessing/classify.py
-4
View File
@@ -5,10 +5,6 @@ import numpy as np
import csv import csv
import os import os
# Disable CUDA
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
model = hub.load('https://tfhub.dev/google/yamnet/1') model = hub.load('https://tfhub.dev/google/yamnet/1')
#Find the name of the class with the top score when mean-aggregated across frames. #Find the name of the class with the top score when mean-aggregated across frames.