AnhP

/

DJCM-Test

like 4

• "torch.nn.modules.linear.Linear",
• "torch.nn.modules.conv.ConvTranspose2d",
• "einops.layers.torch.Rearrange",
• "__main__.DJCM",
• "torch.nn.modules.batchnorm.BatchNorm2d",
• "torch.FloatStorage",
• "torch.nn.modules.activation.PReLU",
• "__main__.PE_Decoder",
• "__main__.BiGRU",
• "__main__.ResDecoderBlock",
• "numpy.ndarray",
• "__main__.Spectrogram",
• "numpy.dtype",
• "__main__.LatentBlocks",
• "torch.nn.modules.rnn.GRU",
• "torch.LongStorage",
• "torch.nn.modules.container.ModuleList",
• "torch.nn.modules.conv.Conv2d",
• "torch.nn.modules.activation.Sigmoid",
• "__builtin__.set",
• "torch.nn.modules.container.Sequential",
• "__main__.STFT",
• "numpy.core.multiarray._reconstruct",
• "torch.device",
• "torch._utils._rebuild_parameter",
• "__main__.Encoder",
• "__main__.ResEncoderBlock",
• "__main__.ResConvBlock",
• "torch.nn.modules.pooling.MaxPool2d",
• "collections.OrderedDict",
• "torch.nn.parallel.data_parallel.DataParallel",
• "torch._utils._rebuild_tensor_v2",
• "_codecs.encode",
• "__main__.Decoder",
• "torch.nn.modules.conv.Conv1d"

How to fix it?