mirror of
https://github.com/facebookresearch/fairseq.git
synced 2024-08-16 20:10:40 +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