Add model card
9b44687 verified - 1.52 kB initial commit
- 7.85 kB Add model card
- 182 Bytes Upload Fast-FoundationStereo v1.0 checkpoint
model_best_bp2_serialize.pth Detected Pickle imports (77)
- "torch.nn.modules.activation.ReLU",
- "torch.nn.modules.linear.Linear",
- "timm.models.edgenext.ConvBlock",
- "core.submodule.ResnetBasicBlock",
- "core.distill_block.ForwardHelper",
- "torch.nn.modules.pooling.AdaptiveMaxPool2d",
- "torch.LongStorage",
- "torch.nn.modules.activation.LeakyReLU",
- "foundation_stereo_ori.submodule.BasicConv_IN",
- "core.submodule.BasicConv_IN",
- "torch.nn.modules.instancenorm.InstanceNorm2d",
- "torch.nn.modules.container.ModuleDict",
- "typing.Any",
- "__builtin__.list",
- "core.extractor.Feature",
- "torch.nn.modules.conv.Conv3d",
- "omegaconf.dictconfig.DictConfig",
- "foundation_stereo_ori.submodule.EdgeNextConvEncoder",
- "torch.nn.modules.conv.ConvTranspose2d",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.nn.modules.conv.Conv2d",
- "timm.models.edgenext.SplitTransposeBlock",
- "_codecs.encode",
- "foundation_stereo_ori.update.BasicSelectiveMultiUpdateBlock",
- "collections.defaultdict",
- "collections.OrderedDict",
- "torch.nn.modules.batchnorm.BatchNorm3d",
- "foundation_stereo_ori.submodule.CostVolumeDisparityAttention",
- "numpy.dtype",
- "foundation_stereo_ori.update.BasicMotionEncoder",
- "core.foundation_stereo.FastFoundationStereo",
- "torch._utils._rebuild_tensor_v2",
- "foundation_stereo_ori.extractor.ContextNetSharedBackbone",
- "torch.float32",
- "timm.layers.norm.LayerNorm2d",
- "foundation_stereo_ori.submodule.SpatialAttentionExtractor",
- "foundation_stereo_ori.submodule.Conv2x",
- "torch.nn.modules.activation.Sigmoid",
- "torch.FloatStorage",
- "torch.nn.modules.dropout.Dropout",
- "torch.HalfStorage",
- "__builtin__.set",
- "timm.models.edgenext.CrossCovarianceAttn",
- "timm.models.edgenext.PositionalEncodingFourier",
- "foundation_stereo_ori.submodule.FeatureAtt",
- "timm.models.edgenext.EdgeNeXtStage",
- "core.distill_block.PostForwardHelper",
- "torch.nn.modules.container.Sequential",
- "core.submodule.Conv2x_IN",
- "omegaconf.base.ContainerMetadata",
- "foundation_stereo_ori.submodule.BasicConv",
- "torch.nn.modules.upsampling.Upsample",
- "torch.CharStorage",
- "foundation_stereo_ori.submodule.ResnetBasicBlock3D",
- "omegaconf.base.Metadata",
- "torch.nn.modules.batchnorm.SyncBatchNorm",
- "torch.nn.modules.pooling.AdaptiveAvgPool2d",
- "torch.nn.modules.conv.ConvTranspose3d",
- "__builtin__.long",
- "foundation_stereo_ori.submodule.PositionalEmbedding",
- "torch._utils._rebuild_parameter",
- "foundation_stereo_ori.update.DispHead",
- "foundation_stereo_ori.submodule.FlashMultiheadAttention",
- "foundation_stereo_ori.submodule.ChannelAttentionEnhancement",
- "__builtin__.dict",
- "foundation_stereo_ori.foundation_stereo.hourglass",
- "foundation_stereo_ori.submodule.Conv3dNormActReduced",
- "torch.nn.modules.linear.Identity",
- "timm.layers.mlp.Mlp",
- "numpy._core.multiarray.scalar",
- "torch.nn.modules.container.ModuleList",
- "foundation_stereo_ori.update.RaftConvGRU",
- "foundation_stereo_ori.submodule.FlashAttentionTransformerEncoderLayer",
- "omegaconf.nodes.AnyNode",
- "omegaconf.listconfig.ListConfig",
- "torch.nn.modules.activation.GELU",
- "foundation_stereo_ori.update.SelectiveConvGRU"
How to fix it?
71.1 MB Upload Fast-FoundationStereo v1.0 checkpoint