update

op.py

@@ -1,25 +0,0 @@
-import torch
-
-# state_dict = torch.load("model_485000.pt", map_location="cpu")
-
-# new_state_dict = state_dict.copy()["ema_model_state_dict"]
-# print(state_dict.keys())
-# print(new_state_dict.keys())
-
-# torch.save(new_state_dict, "model_485000_ema.pt")
-
-ema_state_dict = torch.load("model_485000_ema.pt", map_location="cpu")
-print(ema_state_dict.keys())
-
-for k in list(ema_state_dict.keys()):
-    if "initted" in k or "step" in k or "melody_mel_spec.mel_basis" in k:
-        ema_state_dict.pop(k)
-        continue
-
-    if k.startswith("ema_model."):
-        new_k = k.replace("ema_model.", "")
-        ema_state_dict[new_k] = ema_state_dict.pop(k)
-
-    # remove "initted", "step", "melody_mel_spec.mel_basis".
-
-torch.save(ema_state_dict, "model_485000_ema_fixed.pt")