use InstanceNorm2d instead of LayerNorm

2021-12-07 17:33:37 +01:00 · 2021-12-07 17:33:37 +01:00 · 9325503b51
commit 9325503b51
parent 1e91784a91
1 changed files with 1 additions and 1 deletions
--- a/model/networks.py
+++ b/model/networks.py
@ -193,7 +193,7 @@ class DispNetS(TimedModule):
        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.InstanceNorm2d(out_planes),
            torch.nn.ReLU(inplace=True)
        )