A bug in the residual layer

[mk-0]

Thank you for this great project.
I've been playing with it for some time and noticed that FineADAINResBlock2d (and therefore also FineADAINResBlocks) is just a linear layer with no activation, because line 346 has self.conv2(x), instead of self.conv2(dx)

PIRender/generators/base_function.py

Lines 344 to 348 in 96a157e

	def forward(self, x, z):
	dx = self.actvn(self.norm1(self.conv1(x), z))
	dx = self.norm2(self.conv2(x), z)
	out = dx + x
	return out

[sdulyq]

Thank you