Merge branch '4.1-Stable' into 4.1-Latest

This commit is contained in:
YuriHead 2023-07-09 03:46:23 +08:00 committed by GitHub
commit 3bebce758b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions

View File

@ -99,7 +99,7 @@ def run(rank, n_gpus, hps):
name=utils.latest_checkpoint_path(hps.model_dir, "D_*.pth")
global_step=int(name[name.rfind("_")+1:name.rfind(".")])+1
#global_step = (epoch_str - 1) * len(train_loader)
except: # noqa: E722 I have no idea about this CC: @ylzz1997
except Exception:
print("load old checkpoint failed...")
epoch_str = 1
global_step = 0

View File

@ -165,7 +165,7 @@ def load_checkpoint(checkpoint_path, model, optimizer=None, skip_optimizer=False
# print("load", k)
new_state_dict[k] = saved_state_dict[k]
assert saved_state_dict[k].shape == v.shape, (saved_state_dict[k].shape, v.shape)
except: # noqa: E722 I have no idea about this CC: @ylzz1997
except Exception:
if "enc_q" not in k or "emb_g" not in k:
print("error, %s is not in the checkpoint" % k)
logger.info("%s is not in the checkpoint" % k)
@ -485,7 +485,7 @@ def train_index(spk_name,root_dir = "dataset/44k/"): #from: RVC https://github.
.fit(big_npy)
.cluster_centers_
)
except: # noqa: E722
except Exception:
info = traceback.format_exc()
print(info)
n_ivf = min(int(16 * np.sqrt(big_npy.shape[0])), big_npy.shape[0] // 39)