From 2150fd365f9512d55c78cdbef096b7240f619b21 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Thu, 7 Oct 2021 08:56:52 +0200 Subject: Lint --- text_recognizer/networks/vqvae/vqvae.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/text_recognizer/networks/vqvae/vqvae.py b/text_recognizer/networks/vqvae/vqvae.py index 2d17e0f..d876ca1 100644 --- a/text_recognizer/networks/vqvae/vqvae.py +++ b/text_recognizer/networks/vqvae/vqvae.py @@ -29,9 +29,7 @@ class VQVAE(nn.Module): in_channels=embedding_dim, out_channels=hidden_dim, kernel_size=1 ) self.quantizer = VectorQuantizer( - num_embeddings=num_embeddings, - embedding_dim=embedding_dim, - decay=decay, + num_embeddings=num_embeddings, embedding_dim=embedding_dim, decay=decay, ) def encode(self, x: Tensor) -> Tensor: -- cgit v1.2.3-70-g09d2