update doc

SeanLee97 · Jul 27, 2024 · 6cd6442 · 6cd6442
1 parent 701996a
commit 6cd6442
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/angle_emb/angle.py b/angle_emb/angle.py
@@ -1474,7 +1474,7 @@ def evaluate(self, data: Dataset, batch_size: int = 32, metric: str = 'spearman_
     def truncate_layer(self, layer_index: int):
         """ truncate layer
 
-        :param layer_index: int. layer index to be truncated.
+        :param layer_index: int. layers after layer_index will be truncated.
         :return: self
         """
         if len(self.backbone.encoder.layer) < layer_index: