From 6cd6442902b8ed7e2cbc2440ed83f558c9622743 Mon Sep 17 00:00:00 2001
From: Sean Lee <xmlee97@gmail.com>
Date: Sat, 27 Jul 2024 12:56:01 +0800
Subject: [PATCH] update doc

---
 angle_emb/angle.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/angle_emb/angle.py b/angle_emb/angle.py
index d408300..5ed2394 100644
--- 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: