Skip to content

Commit 01d9114

Browse files
authored
Merge pull request #1161 from IAHispano/formatter/main
chore(format): run black on main
2 parents 5f5a10b + 1ec029e commit 01d9114

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

rvc/train/utils.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,11 +79,13 @@ def load_checkpoint(checkpoint_path, model, optimizer=None, load_opt=1):
7979
optimizer,
8080
checkpoint_dict.get("learning_rate", 0),
8181
checkpoint_dict["iteration"],
82-
checkpoint_dict.get("scaler", {})
82+
checkpoint_dict.get("scaler", {}),
8383
)
8484

8585

86-
def save_checkpoint(model, optimizer, learning_rate, iteration, checkpoint_path, scaler):
86+
def save_checkpoint(
87+
model, optimizer, learning_rate, iteration, checkpoint_path, scaler
88+
):
8789
"""
8890
Save the model and optimizer state to a checkpoint file.
8991

0 commit comments

Comments
 (0)