Commit a4974c4c authored by moto's avatar moto
Browse files

Skip hubert_asr_xlarge TS test on Windows (#1800)

parent 7438f325
import os
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
...@@ -192,6 +194,10 @@ class TestWav2Vec2Model(TorchaudioTestCase): ...@@ -192,6 +194,10 @@ class TestWav2Vec2Model(TorchaudioTestCase):
@finetune_factory_funcs @finetune_factory_funcs
def test_finetune_torchscript(self, factory_func): def test_finetune_torchscript(self, factory_func):
"""Wav2Vec2Model should be scriptable""" """Wav2Vec2Model should be scriptable"""
if factory_func.__name__ == 'hubert_asr_xlarge' and os.name == 'nt':
self.skipTest(
'hubert_asr_xlarge is known to fail on Windows CI. '
'See https://github.com/pytorch/pytorch/issues/65776')
self._test_torchscript(factory_func(num_out=32)) self._test_torchscript(factory_func(num_out=32))
def _test_quantize_smoke_test(self, model): def _test_quantize_smoke_test(self, 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