diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-11-12 23:42:03 +0100 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-11-12 23:42:03 +0100 |
| commit | 8fdb6435e15703fa5b76df19728d905650ee1aef (patch) | |
| tree | be3bec9e5cab4ef7f9d94528d102e57ce9b16c3f /src/text_recognizer/networks/util.py | |
| parent | dc28cbe2b4ed77be92ee8b2b69a20689c3bf02a4 (diff) | |
| parent | 6cb08a110620ee09fe9d8a5d008197a801d025df (diff) | |
Working cnn transformer.
Diffstat (limited to 'src/text_recognizer/networks/util.py')
| -rw-r--r-- | src/text_recognizer/networks/util.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/text_recognizer/networks/util.py b/src/text_recognizer/networks/util.py index b31e640..e2d7955 100644 --- a/src/text_recognizer/networks/util.py +++ b/src/text_recognizer/networks/util.py @@ -24,7 +24,7 @@ def sliding_window( """ unfold = nn.Unfold(kernel_size=patch_size, stride=stride) - # Preform the slidning window, unsqueeze as the channel dimesion is lost. + # Preform the sliding window, unsqueeze as the channel dimesion is lost. c = images.shape[1] patches = unfold(images) patches = rearrange( |