torch hub integration, wavfile input, and train from scratch.

Signed-off-by: Harri Taylor <[email protected]>
harritaylor · Feb 24, 2020 · de6d377 · de6d377
1 parent 9331251
commit de6d377
Showing 1 changed file with 0 additions and 12 deletions.
diff --git a/torchvggish/vggish.py b/torchvggish/vggish.py
@@ -3,13 +3,6 @@
 from torch import hub
 import numpy as np
 from . import vggish_input
-# import vggish_input
-
-
-VGGISH_WEIGHTS = "https://github.com/harritaylor/torchvggish/" \
-                 "releases/download/v0.1/vggish-10086976.pth"
-PCA_PARAMS = "https://github.com/harritaylor/torchvggish/" \
-             "releases/download/v0.1/vggish_pca_params-970ea276.pth"
 
 
 class VGG(nn.Module):
@@ -142,8 +135,3 @@ def _preprocess(self, x, fs):
 
     def _postprocess(self, x):
         return self.pproc.postprocess(x)
-
-
-# if __name__ == "__main__":
-#     model = VGGish()
-#     model.forward("/some/path/to/test/wavfile")