code: def restore(self, save_dir, agent_model_name, qmixer_model_name):
      if not os.path.exists(save_dir):
          os.mkdir(save_dir)
      agent_model_path = os.path.join(save_dir, agent_model_name)
      qmixer_model_path = os.path.join(save_dir, qmixer_model_name)
      self.alg.agent_model.set_state_dict(paddle.load(agent_model_path))
      self.alg.qmixer_model.set_state_dict(
          paddle.load(qmixer_model_path)
      )
      print("restore model successfully!")
  
file_path: PARL/examples/QMIX/qmix_agent.py
imports: import paddle
  import parl
  import numpy
  from utils import AvailableActionsSampler
  import os
label_desc: Use image node_id 3 for calling the QMixAgent obj's underlying member method code with example usage: obj.restore(save_dir, agent_model_name, qmixer_model_name) without return types