From 1209928ccb336c2f6ab3b6c85f5843ad501aca99 Mon Sep 17 00:00:00 2001 From: Shichao Sun Date: Mon, 3 May 2021 14:53:57 +0800 Subject: [PATCH] fix typo --- models/simsiam.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/simsiam.py b/models/simsiam.py index 7db8269..c231f6a 100644 --- a/models/simsiam.py +++ b/models/simsiam.py @@ -39,7 +39,7 @@ def __init__(self, in_dim, hidden_dim=2048, out_dim=2048): ) self.layer3 = nn.Sequential( nn.Linear(hidden_dim, out_dim), - nn.BatchNorm1d(hidden_dim) + nn.BatchNorm1d(out_dim) ) self.num_layers = 3 def set_layers(self, num_layers):