Cpt.Captain/connecting_the_dots
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

use InstanceNorm2d instead of LayerNorm

Browse Source
This commit is contained in:
CptCaptain 2021-12-07 17:33:37 +01:00
parent 1e91784a91
commit 9325503b51
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model/networks.py
View File

@ -193,7 +193,7 @@ class DispNetS(TimedModule):
return torch.nn.Sequential( return torch.nn.Sequential(
torch.nn.Conv2d(in_planes, out_planes, kernel_size=3, padding=1), torch.nn.Conv2d(in_planes, out_planes, kernel_size=3, padding=1),
# TODO try this # TODO try this
torch.nn.LayerNorm(out_planes), torch.nn.InstanceNorm2d(out_planes),
torch.nn.ReLU(inplace=True) torch.nn.ReLU(inplace=True)
) )