Add jit model
e61adf1
cpu_jit.pt
Detected Pickle imports (41)
- "__torch__.torch.nn.modules.conv.___torch_mangle_0.Conv2d",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1.Linear",
- "__torch__.torch.nn.modules.normalization.LayerNorm",
- "__torch__.torch.nn.modules.container.___torch_mangle_12.ModuleList",
- "__torch__.torch.nn.modules.batchnorm.BatchNorm1d",
- "__torch__.subsampling.Conv2dSubsampling",
- "__torch__.torch.nn.modules.activation.ReLU",
- "torch.LongStorage",
- "__torch__.torch.nn.modules.linear.___torch_mangle_4.Linear",
- "__torch__.torch.nn.modules.batchnorm.___torch_mangle_8.BatchNorm1d",
- "torch.FloatStorage",
- "__torch__.conformer.Conformer",
- "__torch__.torch.nn.modules.container.Sequential",
- "__torch__.conformer.ConvolutionModule",
- "__torch__.torch.nn.modules.linear.___torch_mangle_3.Linear",
- "__torch__.conformer.ConformerEncoder",
- "__torch__.torch.nn.modules.linear.___torch_mangle_2.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_11.Sequential",
- "__torch__.torch.nn.modules.dropout.Dropout",
- "__torch__.transformer.TransformerDecoderLayer",
- "__torch__.conformer.Swish",
- "__torch__.torch.nn.modules.conv.Conv2d",
- "__torch__.torch.nn.modules.activation.MultiheadAttention",
- "__torch__.conformer.RelPositionMultiheadAttention",
- "__torch__.torch.nn.modules.linear.Linear",
- "__torch__.torch.nn.modules.container.ModuleList",
- "__torch__.label_smoothing.LabelSmoothingLoss",
- "__torch__.torch.nn.modules.linear.___torch_mangle_10.Linear",
- "__torch__.torch.nn.modules.transformer.TransformerDecoder",
- "__torch__.torch.nn.modules.conv.___torch_mangle_7.Conv1d",
- "collections.OrderedDict",
- "__torch__.conformer.ConformerEncoderLayer",
- "__torch__.conformer.RelPositionalEncoding",
- "__torch__.torch.nn.modules.conv.___torch_mangle_9.Conv1d",
- "__torch__.transformer.PositionalEncoding",
- "__torch__.torch.nn.modules.linear.___torch_mangle_5.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_6.Sequential",
- "__torch__.torch.nn.modules.sparse.Embedding",
- "__torch__.torch.nn.modules.linear._LinearWithBias",
- "torch._utils._rebuild_tensor_v2",
- "__torch__.torch.nn.modules.conv.Conv1d"
How to fix it?
481 MB
Add jit model
-
461 MB
Update model with cer 4.26%