From 7060bd555590a8c4d814126444b2cba24f7c306e Mon Sep 17 00:00:00 2001 From: CptCaptain Date: Mon, 29 Nov 2021 17:54:45 +0100 Subject: [PATCH] add LayerNorm to conv() for testing --- model/networks.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/model/networks.py b/model/networks.py index cd21bae..ea6d8ab 100644 --- a/model/networks.py +++ b/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) )