Commit 9f5fa84b authored by Jeff Hwang's avatar Jeff Hwang Committed by Facebook GitHub Bot
Browse files

Revise VGGish inference pipeline test (#3544)

Summary:
Pull Request resolved: https://github.com/pytorch/audio/pull/3544

Revises VGGish inference pipeline test to support internal testing.

Reviewed By: mthrok

Differential Revision: D48058409

fbshipit-source-id: 045140a0e9d50128d32ef6510bdb2f642a365c83
parent eab8aa74
import unittest
import torchaudio import torchaudio
from torchaudio.prototype.pipelines import VGGISH from torchaudio.prototype.pipelines import VGGISH
def test_vggish(): class VGGishPipelineTest(unittest.TestCase):
def test_vggish(self):
input_sr = VGGISH.sample_rate input_sr = VGGISH.sample_rate
input_proc = VGGISH.get_input_processor() input_proc = VGGISH.get_input_processor()
model = VGGISH.get_model() model = VGGISH.get_model()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment