JulianGuerreiro

/

LayoutFlow

like 2

• "einops.layers.torch.Rearrange",
• "torch.nn.modules.dropout.Dropout",
• "torch.nn.modules.linear.Linear",
• "src.models.BaseGenModel.PositionalEncoding",
• "torch.BoolStorage",
• "torch.nn.modules.container.ModuleList",
• "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
• "torch.FloatStorage",
• "torch.device",
• "__builtin__.getattr",
• "src.utils.distribution_sampler.DistributionSampler",
• "torch.nn.functional.relu",
• "torch._utils._rebuild_tensor_v2",
• "torch.nn.modules.container.Sequential",
• "torch.nn.modules.activation.MultiheadAttention",
• "torch.nn.modules.sparse.Embedding",
• "torch.nn.modules.activation.ReLU",
• "src.models.backbone.layoutdm_backbone.AdaLayerNorm",
• "torch.nn.modules.normalization.LayerNorm",
• "src.models.backbone.layoutdm_backbone.TransformerEncoder",
• "torch.nn.modules.activation.SiLU",
• "collections.OrderedDict",
• "src.models.backbone.layoutdm_backbone.Block",
• "torch._utils._rebuild_parameter",
• "src.models.backbone.layoutdm_backbone.LayoutDMBackbone",
• "torch.optim.adamw.AdamW",
• "functools.partial",
• "__builtin__.set"

How to fix it?