pepecalero

/

TorchscriptSplitModels

like 0

• "__torch__.torch.nn.modules.conv.___torch_mangle_3.Conv2d",
• "__torch__.torch.nn.modules.pooling.AdaptiveAvgPool2d",
• "torch.FloatStorage",
• "__torch__.torch.nn.modules.conv.___torch_mangle_1.Conv2d",
• "__torch__.torch.nn.modules.flatten.Flatten",
• "__torch__.___torch_mangle_56.Part",
• "__torch__.torch.nn.modules.conv.Conv2d",
• "__torch__.torch.nn.modules.conv.___torch_mangle_6.Conv2d",
• "__torch__.torch.nn.modules.container.___torch_mangle_55.Sequential",
• "torch.jit._pickle.build_intlist",
• "__torch__.torch.nn.modules.conv.___torch_mangle_4.Conv2d",
• "__torch__.torch.nn.modules.conv.___torch_mangle_5.Conv2d",
• "__torch__.torch.nn.modules.linear.Linear",
• "__torch__.torch.nn.modules.conv.___torch_mangle_0.Conv2d",
• "__torch__.torch.nn.modules.conv.___torch_mangle_2.Conv2d",
• "__torch__.torch.nn.modules.activation.ReLU",
• "collections.OrderedDict",
• "torch._utils._rebuild_tensor_v2",
• "__torch__.torch.nn.modules.pooling.MaxPool2d"

How to fix it?

• "torch._utils._rebuild_tensor_v2",
• "torch.FloatStorage",
• "__torch__.torch.nn.modules.linear.___torch_mangle_7.Linear",
• "__torch__.torch.nn.modules.flatten.Flatten",
• "collections.OrderedDict",
• "__torch__.torch.nn.modules.linear.___torch_mangle_8.Linear",
• "__torch__.___torch_mangle_18.Part",
• "__torch__.torch.nn.modules.container.___torch_mangle_17.Sequential",
• "__torch__.torch.nn.modules.activation.ReLU",
• "__torch__.torch.nn.modules.dropout.Dropout"

How to fix it?