mirror of
https://github.com/facebookresearch/fairseq.git
synced 2024-09-11 17:25:31 +03:00
Add regression test for SimulConvTransformerModel (#3031)
Summary: Pull Request resolved: https://github.com/fairinternal/fairseq-py/pull/3031 Reviewed By: kahne Differential Revision: D34018108 Pulled By: sravyapopuri388 fbshipit-source-id: 4db96653658a998b15c0cdbc2e588198d951a420
This commit is contained in:
parent
cfc4d8475c
commit
67eaecd2fc
30
tests/speech/test_convtransformer_simul_trans.py
Normal file
30
tests/speech/test_convtransformer_simul_trans.py
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
# Copyright (c) Facebook, Inc. and its affiliates.
|
||||||
|
#
|
||||||
|
# This source code is licensed under the MIT license found in the
|
||||||
|
# LICENSE file in the root directory of this source tree.
|
||||||
|
|
||||||
|
import unittest
|
||||||
|
from tests.speech import TestFairseqSpeech
|
||||||
|
|
||||||
|
S3_BASE_URL = "https://dl.fbaipublicfiles.com/fairseq/"
|
||||||
|
|
||||||
|
|
||||||
|
class TestConvtransformerSimulTrans(TestFairseqSpeech):
|
||||||
|
def setUp(self):
|
||||||
|
self._set_up(
|
||||||
|
"simul",
|
||||||
|
"speech_tests/simul",
|
||||||
|
["config_gcmvn_specaug.yaml", "dict.txt", "dev.tsv"],
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_waitk_checkpoint(self):
|
||||||
|
"""Only test model loading since fairseq currently doesn't support inference of simultaneous models"""
|
||||||
|
_, _, _, _ = self.download_and_load_checkpoint(
|
||||||
|
"checkpoint_best.pt",
|
||||||
|
arg_overrides={"config_yaml": "config_gcmvn_specaug.yaml"},
|
||||||
|
)
|
||||||
|
return
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
unittest.main()
|
Loading…
Reference in New Issue
Block a user