fix: wrong parameter name for ClassifHead

meilame-tayebjee · meilame-tayebjee · commit 6e3f39201af0 · 2025-11-25T23:08:45.000+01:00
diff --git 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`!
 
+

Original file line number	Diff line number	Diff line change
`@@ -270,7 +270,7 @@ custom_head = nn.Sequential(`
`270`	`270`	`nn.Linear(64, 5)`
`271`	`271`	`)`
`272`	`272`
`273`		`-head = ClassificationHead(linear=custom_head)`
	`273`	`+head = ClassificationHead(net=custom_head)`
`274`	`274`	```
`275`	`275`
`276`	`276`	`## Complete Architecture`
`@@ -611,3 +611,4 @@ torchTextClassifiers provides a component-based pipeline for text classifica`
`611`	`611`
`612`	`612`	Ready to build your classifier? Start with {doc}`../getting_started/quickstart`!
`613`	`613`
	`614`	`+`