code: def test_torch_wo_auto_param_call(
      driver,
      device,
      n_epochs=2,
  ):
      if driver == "torch_fsdp" and not _TORCH_GREATER_EQUAL_1_12:
          pytest.skip("fsdp 需要 torch 在 1.12 及以上")
  
      model = TorchNormalModel_Classification_3(
          num_labels=NormalClassificationTrainTorchConfig.num_labels,
          feature_dimension=NormalClassificationTrainTorchConfig.feature_dimension,
      )
      optimizers = SGD(model.parameters(), lr=0.001)
      dataset = TorchNormalDataset_Classification(
          num_labels=NormalClassificationTrainTorchConfig.num_labels,
          feature_dimension=NormalClassificationTrainTorchConfig.feature_dimension,
          each_label_data=NormalClassificationTrainTorchConfig.each_label_data,
          seed=NormalClassificationTrainTorchConfig.seed,
      )
      train_dataloader = DataLoader(
          dataset=dataset,
          batch_size=NormalClassificationTrainTorchConfig.batch_size,
          shuffle=True,
      )
  
      trainer = Trainer(
          model=model,
          driver=driver,
          device=device,
          optimizers=optimizers,
          train_dataloader=train_dataloader,
          n_epochs=n_epochs,
          model_wo_auto_param_call=True,
          output_from_new_proc="all",
      )
  
      trainer.run()
  
      if dist.is_initialized():
          dist.destroy_process_group()
  
imports: import os.path
  import subprocess
  import sys
  import pytest
  from dataclasses import dataclass
  from typing import Any
  from pathlib import Path
  from fastNLP.core.controllers.trainer import Trainer
  from tests.helpers.models.torch_model import TorchNormalModel_Classification_1, TorchNormalModel_Classification_3
  from tests.helpers.datasets.torch_data import TorchNormalDataset_Classification
  from tests.helpers.callbacks.helper_callbacks import RecordLossCallback
  from tests.helpers.callbacks.helper_callbacks_torch import RecordAccumulationStepsCallback_Torch
  from tests.helpers.utils import magic_argv_env_context, Capturing
  from fastNLP.envs.distributed import rank_zero_rm
  from fastNLP.envs.imports import _NEED_IMPORT_TORCH, _TORCH_GREATER_EQUAL_1_12
label_desc: Use image node_id 9 for calling a global function with example usage: test_torch_wo_auto_param_call(driver, device, n_epochs) without return types