class_docstr: 测试多卡情况下 save 和 load 相关函数的表现
class_docstr_tok: ["\u6d4b\u8bd5\u591a\u5361\u60c5\u51b5\u4e0b","save","\u548c","load","\u76f8\u5173\u51fd\u6570\u7684\u8868\u73b0"]
code: def setup_class(cls):
      # 不在这里 setup 的话会报错
      cls.driver = generate_driver(10, 10, device=[0, 1])
  
imports: import pytest
  from pathlib import Path
  from fastNLP.core.drivers.paddle_driver.fleet import PaddleFleetDriver
  from fastNLP.core.samplers import RandomSampler, UnrepeatedSampler, BucketedBatchSampler, UnrepeatedRandomSampler, UnrepeatedSequentialSampler
  from tests.helpers.models.paddle_model import PaddleNormalModel_Classification_1
  from tests.helpers.datasets.paddle_data import PaddleNormalDataset, PaddleNormalXYDataset
  from tests.helpers.utils import magic_argv_env_context, recover_logger
  from fastNLP.envs.distributed import rank_zero_rm
  from fastNLP import prepare_paddle_dataloader
  from fastNLP.core.drivers.paddle_driver.dist_utils import fastnlp_paddle_all_gather
  from fastNLP.envs.imports import _NEED_IMPORT_PADDLE
  from fastNLP import logger
label_desc: Use image node_id 1 for calling the TestSaveLoad obj's underlying member method code with example usage: obj.setup_class(cls) without return types