Add missing LeakyReLU after ConvBlock in Discriminator fromRGB (#54)

srihari-humbarwadi · web-flow · commit cdd9002ad171 · 2021-01-04T17:37:22.000+01:00
* added LeakyReLU after ConvBlock in `self.from_rgb`

* fix formatting
diff --git a/pro_gan_pytorch/networks.py b/pro_gan_pytorch/networks.py
@@ -12,7 +12,7 @@
     GenInitialBlock,
 )
 from torch import Tensor
-from torch.nn import Conv2d, ModuleList
+from torch.nn import Conv2d, LeakyReLU, ModuleList, Sequential
 from torch.nn.functional import avg_pool2d, interpolate
 
 
@@ -160,7 +160,10 @@ def __init__(
         self.from_rgb = ModuleList(
             reversed(
                 [
-                    ConvBlock(num_channels, nf(stage), kernel_size=(1, 1))
+                    Sequential(
+                        ConvBlock(num_channels, nf(stage), kernel_size=(1, 1)),
+                        LeakyReLU(0.2)
+                        )
                     for stage in range(1, depth)
                 ]
             )

Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@`
`12`	`12`	`GenInitialBlock,`
`13`	`13`	`)`
`14`	`14`	`from torch import Tensor`
`15`		`-from torch.nn import Conv2d, ModuleList`
	`15`	`+from torch.nn import Conv2d, LeakyReLU, ModuleList, Sequential`
`16`	`16`	`from torch.nn.functional import avg_pool2d, interpolate`
`17`	`17`
`18`	`18`
`@@ -160,7 +160,10 @@ def __init__(`
`160`	`160`	`self.from_rgb = ModuleList(`
`161`	`161`	`reversed(`
`162`	`162`	`[`
`163`		`- ConvBlock(num_channels, nf(stage), kernel_size=(1, 1))`
	`163`	`+ Sequential(`
	`164`	`+ ConvBlock(num_channels, nf(stage), kernel_size=(1, 1)),`
	`165`	`+ LeakyReLU(0.2)`
	`166`	`+ )`
`164`	`167`	`for stage in range(1, depth)`
`165`	`168`	`]`
`166`	`169`	`)`