From 314f9c7597808918d2a1abef18aa5d13bc2eb984 Mon Sep 17 00:00:00 2001 From: Harrison Taylor Date: Thu, 26 Sep 2019 18:39:59 +0100 Subject: [PATCH] Added github-hosted weights --- torchvggish.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/torchvggish.py b/torchvggish.py index b2761ba..7cfa8f0 100644 --- a/torchvggish.py +++ b/torchvggish.py @@ -8,10 +8,8 @@ import errno from urllib.parse import urlparse -# TODO: Upload weights and PCA to project release - -VGGISH_WEIGHTS = "https://users.cs.cf.ac.uk/taylorh23/pytorch/models/vggish-10086976.pth" -PCA_PARAMS = "https://users.cs.cf.ac.uk/taylorh23/pytorch/models/vggish_pca_params-4d878af3.npz" +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-4d878af3.npz" class VGG(nn.Module):