Skip to content

Commit 95073b9

Browse files
Remove steps in low-level documentation
1 parent 8514a8d commit 95073b9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

doc/guides/using_low_level.rst

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ Detailed Guides
121121
# Random task labels
122122
task_labels = torch.randint(0, n_tasks, (batch_size,))
123123
124-
**Step 3: Build a Concept Bottleneck Model**
124+
**Build a Concept Bottleneck Model**
125125

126126
Use a ModuleDict to combine encoder and predictor:
127127

@@ -143,7 +143,7 @@ Detailed Guides
143143
.. dropdown:: Inference and Training
144144
:icon: rocket
145145

146-
**Step 1: Inference**
146+
**Inference**
147147

148148
Once a concept bottleneck model is built, we can perform inference by first obtaining
149149
concept activations from the encoder, and then task predictions from the predictor:
@@ -159,7 +159,7 @@ Detailed Guides
159159
print(f"Concept endogenous shape: {concept_endogenous.shape}") # [32, 5]
160160
print(f"Task endogenous shape: {task_endogenous.shape}") # [32, 3]
161161
162-
**Step 2: Compute Loss and Train**
162+
**Compute Loss and Train**
163163

164164
Train with both concept and task supervision:
165165

0 commit comments

Comments
 (0)