add LayerNorm to conv() for testing

model/networks.py

@@ -192,6 +192,8 @@ class DispNetS(TimedModule):
     def conv(self, in_planes, out_planes):
         return torch.nn.Sequential(
             torch.nn.Conv2d(in_planes, out_planes, kernel_size=3, padding=1),
+            # TODO try this
+            torch.nn.LayerNorm(out_planes),
             torch.nn.ReLU(inplace=True)
         )