fixed typo. self.preprocess → self.postprocess

harritaylor · Mar 19, 2020 · 547c94c · 547c94c
1 parent b229b32
commit 547c94c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvggish/vggish.py b/torchvggish/vggish.py
@@ -149,7 +149,7 @@ def __init__(self, urls, pretrained=True, preprocess=True, postprocess=True, pro
 
         self.preprocess = preprocess
         self.postprocess = postprocess
-        if self.preprocess:
+        if self.postprocess:
             self.pproc = Postprocessor()
             if pretrained:
                 state_dict = hub.load_state_dict_from_url(urls['pca'], progress=progress)