Bugfix in the example CNN

Author: davda54

example/model/wide_res_net.py

@@ -19,7 +19,7 @@ def __init__(self, channels: int, dropout: float):
         ]))
 
     def forward(self, x):
-        return self.block(x)
+        return x + self.block(x)
 
 
 class DownsampleUnit(nn.Module):
@@ -90,4 +90,4 @@ def _initialize(self):
                 m.bias.data.zero_()
 
     def forward(self, x):
-        return self.f(x)
+        return self.f(x)