howard9199

/

tmp

like 0

• "collections.OrderedDict",
• "torch.nn.modules.dropout.Dropout",
• "torch.FloatStorage",
• "image_common.network.UNet",
• "image_common.module.TimeEmbedding",
• "torch._utils._rebuild_tensor_v2",
• "__builtin__.set",
• "torch.nn.modules.linear.Identity",
• "torch.nn.modules.sparse.Embedding",
• "image_common.module.Swish",
• "torch.nn.modules.container.ModuleList",
• "image_common.module.AttnBlock",
• "torch.nn.modules.activation.SiLU",
• "image_common.fm.FMScheduler",
• "torch.nn.modules.normalization.GroupNorm",
• "image_common.module.UpSample",
• "torch.nn.modules.conv.Conv2d",
• "torch._utils._rebuild_parameter",
• "torch.nn.modules.container.Sequential",
• "image_common.module.ResBlock",
• "image_common.module.DownSample",
• "torch.nn.modules.linear.Linear"

How to fix it?