From 1bccf71cf4eec335001b50a8fbc0c991d0e6d13a Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Wed, 4 Aug 2021 22:15:36 +0200
Subject: Add conv attention, up and downsampling to vqvae module

---
 text_recognizer/networks/vqvae/quantizer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'text_recognizer/networks/vqvae/quantizer.py')

diff --git a/text_recognizer/networks/vqvae/quantizer.py b/text_recognizer/networks/vqvae/quantizer.py
index 1b59e78..a4f11f0 100644
--- a/text_recognizer/networks/vqvae/quantizer.py
+++ b/text_recognizer/networks/vqvae/quantizer.py
@@ -7,7 +7,7 @@ from einops import rearrange
 import torch
 from torch import nn
 from torch import Tensor
-from torch.nn import functional as F
+import torch.nn.functional as F
 
 
 class EmbeddingEMA(nn.Module):
-- 
cgit v1.2.3-70-g09d2