Duplicated from  xinghaochen/tinysam

Supbatomic

/

tinysam

like 0

• "mobile_sam.modeling.tiny_vit_sam.Attention",
• "torch.nn.modules.linear.Identity",
• "mobile_sam.modeling.tiny_vit_sam.Mlp",
• "mobile_sam.modeling.tiny_vit_sam.Conv2d_BN",
• "torch.nn.modules.conv.Conv2d",
• "torch.nn.modules.activation.GELU",
• "torch.nn.modules.dropout.Dropout",
• "mobile_sam.modeling.tiny_vit_sam.ConvLayer",
• "torch.nn.modules.container.Sequential",
• "torch._utils._rebuild_parameter_with_state",
• "torch.HalfStorage",
• "mobile_sam.modeling.tiny_vit_sam.TinyViTBlock",
• "torch.nn.modules.linear.Linear",
• "torch._utils._rebuild_parameter",
• "mobile_sam.modeling.tiny_vit_sam.TinyViT",
• "mobile_sam.modeling.tiny_vit_sam.PatchMerging",
• "mobile_sam.modeling.tiny_vit_sam.PatchEmbed",
• "collections.OrderedDict",
• "torch.nn.modules.batchnorm.BatchNorm2d",
• "torch.nn.modules.container.ModuleList",
• "torch._utils._rebuild_tensor_v2",
• "mobile_sam.modeling.tiny_vit_sam.MBConv",
• "__builtin__.set",
• "torch.nn.modules.normalization.LayerNorm",
• "torch.LongStorage",
• "mobile_sam.modeling.tiny_vit_sam.BasicLayer",
• "mobile_sam.modeling.tiny_vit_sam.LayerNorm2d"

How to fix it?

• "collections.OrderedDict",
• "torch.QInt8Storage",
• "torch.FloatStorage",
• "torch.per_tensor_affine",
• "torch.LongStorage",
• "torch._utils._rebuild_tensor_v2",
• "torch._utils._rebuild_qtensor",
• "torch.qint8",
• "torch._utils._rebuild_parameter"

How to fix it?