PrunaAI

/

maxvit_base_tf_384.in21k_ft_in1k-turbo-tiny-green-smashed

like 0

• "timm.layers.norm.LayerNorm2d",
• "collections.OrderedDict",
• "dill._dill._import_module",
• "torch.nn.modules.dropout.Dropout",
• "timm.layers.activations.Tanh",
• "typing.Optional",
• "timm.layers.padding.pad_same_arg",
• "timm.models.maxxvit.MaxxVitBlock",
• "pit.half.half.HalfHelper",
• "dill._dill",
• "typing.Union",
• "torch.nn.modules.container.Sequential",
• "_frozen_importlib_external.SourceFileLoader",
• "builtins.setattr",
• "timm.layers.squeeze_excite.SEModule",
• "torch._utils._rebuild_tensor_v2",
• "timm.layers.activations.GELUTanh",
• "dill._dill._create_code",
• "timm.layers.mlp.Mlp",
• "timm.layers.norm_act.BatchNormAct2d",
• "timm.models.maxxvit.MaxxVitStage",
• "torch.nn.modules.pooling.AdaptiveAvgPool2d",
• "timm.models.maxxvit.AttentionCl",
• "_operator.getitem",
• "timm.models.maxxvit.Stem",
• "timm.layers.padding.get_padding_value",
• "torch.nn.modules.linear.Linear",
• "torch.nn.modules.flatten.Flatten",
• "timm.models.maxxvit.MaxxVit",
• "timm.layers.classifier.NormMlpClassifierHead",
• "timm.layers.padding.pad_same",
• "builtins.__dict__",
• "_frozen_importlib.ModuleSpec",
• "timm.layers.pool2d_same.AvgPool2dSame",
• "timm.layers.pos_embed_rel.RelPosBiasTf",
• "torch.nn.modules.linear.Identity",
• "timm.layers.adaptive_avgmax_pool.SelectAdaptivePool2d",
• "torch.nn.modules.conv.Conv2d",
• "timm.layers.config.is_scriptable",
• "timm.models.maxxvit.Downsample2d",
• "timm.layers.norm.LayerNorm",
• "timm.models.maxxvit.MbConvBlock",
• "timm.models.maxxvit.PartitionAttentionCl",
• "dill._dill._create_function",
• "torch.Tensor",
• "dill._dill._load_type",
• "dill._dill._create_cell",
• "typing.Tuple",
• "builtins.getattr",
• "torch._utils._rebuild_parameter",
• "timm.layers.config.is_exportable",
• "torch.storage._load_from_bytes",
• "torch.nn.modules.activation.SiLU",
• "dill._dill._create_type",
• "timm.models.maxxvit.MaxxVit.forward",
• "timm.layers.activations.Sigmoid",
• "pit.x_fast.compile.X_Fast_Compiler"

How to fix it?