search

taoxugit / AttnGAN

MIT License
1.33k stars 420 forks source link

upblock #69

Closed lsabrinax closed 4 years ago

lsabrinax commented 4 years ago 
def upBlock(in_planes, out_planes):
    block = nn.Sequential(
        nn.Upsample(scale_factor=2, mode='nearest'),
        conv3x3(in_planes, out_planes * 2),
        nn.BatchNorm2d(out_planes * 2),
        GLU())
    return block

i think the out_planes shouldn't times 2, and then the channel can decrease step by step ,because the argument feed into the function in_planes have been two times than out_planes

lsabrinax commented 4 years ago

this is my fault. the code has no problem. I have got it