Comment edit

neuro-galaxy · Nov 28, 2024 · 6cd4134 · 6cd4134
1 parent 69e2900
commit 6cd4134
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)