Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
kaldi_compatibility_cpu_test.py 329 Bytes
Newer Older
moto's avatar
Get rid of dynamic test suite generation (#716)  
moto committed
1
2 
import torch
moto's avatar
Apply tweaks required for running test in fbcode (#663)  
moto committed
3
4 
from . import common_utils
from .kaldi_compatibility_impl import Kaldi
moto's avatar
Adopt PyTorch's test utility on kaldi compatibility test (#650)  
moto committed
5
6
moto's avatar
Make TestCases backend-aware (#719)  
moto committed
7 
class TestKaldiFloat32(Kaldi, common_utils.PytorchTestCase):
moto's avatar
Get rid of dynamic test suite generation (#716)  
moto committed
8
9
10
11 
    dtype = torch.float32
    device = torch.device('cpu')
moto's avatar
Make TestCases backend-aware (#719)  
moto committed
12 
class TestKaldiFloat64(Kaldi, common_utils.PytorchTestCase):
moto's avatar
Get rid of dynamic test suite generation (#716)  
moto committed
13
14 
    dtype = torch.float64
    device = torch.device('cpu')