From 9fd6c561d67e8eabca56b24e95a56dba4351afe7 Mon Sep 17 00:00:00 2001
From: Meilame Tayebjee <114609737+meilame-tayebjee@users.noreply.github.com>
Date: Tue, 25 Nov 2025 23:08:27 +0100
Subject: [PATCH] fix: wrong parameter name for ClassifHead

---
 docs/source/architecture/overview.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/docs/source/architecture/overview.md b/docs/source/architecture/overview.md
index ae1538b..06b4467 100644
--- a/docs/source/architecture/overview.md
+++ b/docs/source/architecture/overview.md
@@ -270,7 +270,7 @@ custom_head = nn.Sequential(
     nn.Linear(64, 5)
 )
 
-head = ClassificationHead(linear=custom_head)
+head = ClassificationHead(net=custom_head)
 ```
 
 ## Complete Architecture
@@ -611,3 +611,4 @@ torchTextClassifiers provides a **component-based pipeline** for text classifica
 
 Ready to build your classifier? Start with {doc}`../getting_started/quickstart`!
 
+