update

examples/mpnet_aishell/step_2_train_model.py

@@ -204,11 +204,11 @@ def main():
         logger.info(f"load state dict for optim_g.")
         with open(optim_g_pth.as_posix(), "rb") as f:
             state_dict = torch.load(f, map_location="cpu", weights_only=True)
-        optim_g.load_state_dict(state_dict, strict=True)
+        optim_g.load_state_dict(state_dict)
         logger.info(f"load state dict for optim_d.")
         with open(optim_d_pth.as_posix(), "rb") as f:
             state_dict = torch.load(f, map_location="cpu", weights_only=True)
-        optim_d.load_state_dict(state_dict, strict=True)
+        optim_d.load_state_dict(state_dict)
 
     scheduler_g = torch.optim.lr_scheduler.ExponentialLR(optim_g, gamma=config.lr_decay, last_epoch=last_epoch)
     scheduler_d = torch.optim.lr_scheduler.ExponentialLR(optim_d, gamma=config.lr_decay, last_epoch=last_epoch)