nvidia

/

c-fast-foundationstereo

like 1

Follow

NVIDIA 61.3k

• "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?