hwd-code

self.Decoder_inchannel = [1024, 2048, 1024, 512, 256, 128] self.Decoder_outchannel = [1024, 512, 256, 128, 64, 32] self.Decoder = nn.ModuleDict({f'layer_{i}' : nn.Sequential( nn.ConvTranspose2d(self.Decoder_inchannel[i], self.Decoder_outchannel[i], kernel_size=4, stride=2, padding=1), nn.BatchNorm2d(self.Decoder_outchannel[i]), nn.LeakyReLU(0.1) )for i...