disable cuda for classify.py · d0f6c8c77a - freqsplit - Gitea: Git with a cup of tea

joel/freqsplit

disable cuda for classify.py

This commit is contained in:

Joel Mathew Thomas

2024-12-26 16:30:19 +05:30

parent 1c71419537

commit d0f6c8c77a

1 changed files with 5 additions and 0 deletions

									
										src/preprocessing/classify.py
									
		+5
		
												View File
												
					@@ -3,6 +3,11 @@ import tensorflow_hub as hub

					import librosa

					import librosa

					import numpy as np

					import numpy as np

					import csv

					import csv

					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')