From 6cd413429c1435d17b8e491b2512e6ba8bb8f497 Mon Sep 17 00:00:00 2001 From: Vinam Arora Date: Thu, 28 Nov 2024 02:20:24 -0500 Subject: [PATCH] Comment edit --- torch_brain/models/poyo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_brain/models/poyo.py b/torch_brain/models/poyo.py index ee554d0..4dae68d 100644 --- a/torch_brain/models/poyo.py +++ b/torch_brain/models/poyo.py @@ -33,7 +33,7 @@ class POYO(nn.Module): Args: dim: Hidden dimension of the model - dim_out: Dimension of the decoding targets + dim_out: Dimension of the output dim_head: Dimension of each attention head num_latents: Number of unique latent tokens (repeated at every latent step) depth: Number of processing layers (self-attentions in the latent space)