from local

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

View File

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