Clean and fix docs

2024-03-11 17:05:42 +09:00
parent dada0af2b6
commit 44851a6219
6 changed files with 13 additions and 97 deletions
--- a/style_bert_vits2/models/utils/safetensors.py
+++ b/style_bert_vits2/models/utils/safetensors.py
@@ -22,7 +22,7 @@ def load_safetensors(
        for_infer (bool): 推論用に読み込むかどうかのフラグ

    Returns:
-        tuple[torch.nn.Module, Optional[int]]: 読み込まれたモデルとイテレーション番号（存在する場合）
+        tuple[torch.nn.Module, Optional[int]]: 読み込まれたモデルとイテレーション回数（存在する場合）
    """

    tensors: dict[str, Any] = {}
@@ -64,7 +64,7 @@ def save_safetensors(

    Args:
        model (torch.nn.Module): 保存するモデル
-        iteration (int): イテレーション番号
+        iteration (int): イテレーション回数
        checkpoint_path (Union[str, Path]): 保存先のパス
        is_half (bool): モデルを半精度で保存するかどうかのフラグ
        for_infer (bool): 推論用に保存するかどうかのフラグ