From 9325503b515321ccc61e379f5cce34c1bb0e1521 Mon Sep 17 00:00:00 2001 From: CptCaptain Date: Tue, 7 Dec 2021 17:33:37 +0100 Subject: [PATCH] use InstanceNorm2d instead of LayerNorm --- model/networks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model/networks.py b/model/networks.py index ea6d8ab..a40c527 100644 --- 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) )