update

examples/mpnet_aishell/step_2_train_model.py

@@ -182,11 +182,11 @@ def main():
 
         logger.info(f"load state dict for generator.")
         with open(generator_pt.as_posix(), "rb") as f:
-            state_dict = torch.load(f, map_location="cpu")
+            state_dict = torch.load(f, map_location="cpu", weights_only=True)
         generator.load_state_dict(state_dict, strict=True)
         logger.info(f"load state dict for discriminator.")
         with open(discriminator_pt.as_posix(), "rb") as f:
-            state_dict = torch.load(f, map_location="cpu")
+            state_dict = torch.load(f, map_location="cpu", weights_only=True)
         discriminator.load_state_dict(state_dict, strict=True)
 
     # optimizer