jlin404

/

squeezeformer_diff

Like 0

• "__torch__.torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_0.Linear",
• "__torch__.torch.nn.modules.conv.___torch_mangle_2.Conv1d",
• "torch._utils._rebuild_tensor_v2",
• "__torch__.torch.nn.modules.linear.Linear",
• "__torch__.layers.GLU",
• "__torch__.torch.nn.modules.linear.___torch_mangle_8.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1.Linear",
• "__torch__.torch.nn.modules.container.ModuleList",
• "torch.FloatStorage",
• "__torch__.layers.TransformerEncoder",
• "__torch__.layers.ECA",
• "__torch__.torch.nn.modules.conv.Conv1d",
• "__torch__.squeezeformer.Squeezeformer",
• "__torch__.torch.nn.modules.linear.___torch_mangle_4.Linear",
• "__torch__.layers.ScaleBias",
• "__torch__.torch.nn.modules.batchnorm.BatchNorm1d",
• "__torch__.layers.GLUMlp",
• "__torch__.torch.nn.modules.container.___torch_mangle_3.ModuleList",
• "torch.LongStorage",
• "__torch__.torch.nn.modules.linear.___torch_mangle_5.Linear",
• "__torch__.layers.HeadDense",
• "__torch__.torch.nn.modules.activation.MultiheadAttention",
• "torch.jit._pickle.build_intlist",
• "collections.OrderedDict",
• "__torch__.torch.nn.modules.normalization.LayerNorm",
• "__torch__.layers.Conv1DBlockSqueezeformer",
• "__torch__.torch.nn.modules.linear.___torch_mangle_6.Linear",
• "__torch__.layers.___torch_mangle_7.GLUMlp",
• "__torch__.torch.nn.modules.activation.SiLU",
• "__torch__.wrap_model.WrappedModel",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist",
• "torch.jit._pickle.build_intlist"

How to fix it?