from local

This commit is contained in:
dohe0342 2023-02-02 13:26:57 +09:00
parent bdf46c269e
commit 7acbff11f0
2 changed files with 1 additions and 0 deletions

View File

@ -28,6 +28,7 @@ import torch.multiprocessing as mp
import torch.nn as nn import torch.nn as nn
from asr_datamodule import AishellAsrDataModule from asr_datamodule import AishellAsrDataModule
from conformer import Conformer from conformer import Conformer
from transformer import Transformer
from lhotse.utils import fix_random_seed from lhotse.utils import fix_random_seed
from torch.nn.parallel import DistributedDataParallel as DDP from torch.nn.parallel import DistributedDataParallel as DDP
from torch.nn.utils import clip_grad_norm_ from torch.nn.utils import clip_grad_norm_