From 764a14468ab8b669dd0906641b891b9bdec7b96a Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 13 Sep 2022 18:46:25 +0200 Subject: Rename hidden dim to dim --- text_recognizer/networks/transformer/embeddings/fourier.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) (limited to 'text_recognizer/networks/transformer') diff --git a/text_recognizer/networks/transformer/embeddings/fourier.py b/text_recognizer/networks/transformer/embeddings/fourier.py index ade589c..7843c60 100644 --- a/text_recognizer/networks/transformer/embeddings/fourier.py +++ b/text_recognizer/networks/transformer/embeddings/fourier.py @@ -8,12 +8,10 @@ from torch import Tensor class PositionalEncoding(nn.Module): """Encodes a sense of distance or time for transformer networks.""" - def __init__( - self, hidden_dim: int, dropout_rate: float, max_len: int = 1000 - ) -> None: + def __init__(self, dim: int, dropout_rate: float, max_len: int = 1000) -> None: super().__init__() self.dropout = nn.Dropout(p=dropout_rate) - pe = self.make_pe(hidden_dim, max_len) + pe = self.make_pe(dim, max_len) self.register_buffer("pe", pe) @staticmethod -- cgit v1.2.3-70-g09d2