from local

This commit is contained in:
dohe0342 2023-02-02 13:56:56 +09:00
parent 078a12763f
commit e657ab5ae3
2 changed files with 1 additions and 0 deletions

View File

@ -24,6 +24,7 @@ from label_smoothing import LabelSmoothingLoss
from subsampling import Conv2dSubsampling, VggSubsampling from subsampling import Conv2dSubsampling, VggSubsampling
from torch.nn.utils.rnn import pad_sequence from torch.nn.utils.rnn import pad_sequence
from torch.nn.modules import Module from torch.nn.modules import Module
import torch.Tensor as Tensor
# Note: TorchScript requires Dict/List/etc. to be fully typed. # Note: TorchScript requires Dict/List/etc. to be fully typed.
Supervisions = Dict[str, torch.Tensor] Supervisions = Dict[str, torch.Tensor]