fix dependancy problems, removed sound package

DTaoo · Jul 27, 2018 · 23ec031 · 23ec031
1 parent 57c2e0f
commit 23ec031
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/vggish.py b/vggish.py
@@ -19,12 +19,11 @@
 from keras.engine.topology import get_source_inputs
 from keras import backend as K
 
-from sound import vggish_params2 as params
+import vggish_params as params
 
 
 # weight path
-WEIGHTS_PATH = '/mount/hudi/moe/moce/soundnet_moce/sound/weights/vggish_audioset_weights_without_fc2.h5'
-WEIGHTS_PATH_TOP = '/mount/hudi/moe/moce/soundnet_moce/sound/weights/vggish_audioset_weights.h5'
+WEIGHTS_PATH = '/home/brain/Documents/git/VGGish/vggish_audioset_weights_without_fc2.h5'
 
 def VGGish(load_weights=True, weights='audioset',
            input_tensor=None, input_shape=None,