From 59b3520616e3c2b50c9c9c927d6f8f0ddf2d233c Mon Sep 17 00:00:00 2001 From: David Ruddell Date: Mon, 1 Aug 2022 01:51:22 -0400 Subject: [PATCH] fix: typo in _build_project method 'rearrage' -> 'rearrange' in name of sequential projection layer --- lib/models/cls_cvt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lib/models/cls_cvt.py b/lib/models/cls_cvt.py index 0267176..e5952e6 100644 --- a/lib/models/cls_cvt.py +++ b/lib/models/cls_cvt.py @@ -140,7 +140,7 @@ def _build_projection(self, groups=dim_in )), ('bn', nn.BatchNorm2d(dim_in)), - ('rearrage', Rearrange('b c h w -> b (h w) c')), + ('rearrange', Rearrange('b c h w -> b (h w) c')), ])) elif method == 'avg': proj = nn.Sequential(OrderedDict([ @@ -150,7 +150,7 @@ def _build_projection(self, stride=stride, ceil_mode=True )), - ('rearrage', Rearrange('b c h w -> b (h w) c')), + ('rearrange', Rearrange('b c h w -> b (h w) c')), ])) elif method == 'linear': proj = None