class_docstr: 用于执行 ``oneflow`` 动态图 cpu 和 单卡 gpu 运算的 ``driver``。
  
  :param model: 传入给 ``Trainer`` 的 ``model`` 参数
  :param device: oneflow.device，当前进程所使用的设备
  :param fp16: 是否开启 fp16；目前动态图的单卡下该参数无效。
  :param oneflow_kwargs:
  :kwargs:
      * *model_wo_auto_param_call* (``bool``) -- 是否关闭在训练时调用我们的 ``auto_param_call`` 函数来自动匹配 batch 和前向函数的参数的行为。
  
      .. note::
  
          关于该参数的详细说明，请参见 :class:`~fastNLP.core.controllers.Trainer` 中的描述；函数 ``auto_param_call`` 详见 :func:`fastNLP.core.utils.auto_param_call`。
docstr: 将模型迁移到相应的设备上。
docstr_tok: ["\u5c06\u6a21\u578b\u8fc1\u79fb\u5230\u76f8\u5e94\u7684\u8bbe\u5907\u4e0a\u3002"]
code: def setup(self):
      r"""
      将模型迁移到相应的设备上。
      """
      if self.model_device is not None:
          self.model.to(self.model_device)
  
file_path: fastNLP/fastNLP/core/drivers/oneflow_driver/single_device.py
imports: import os
  from typing import Dict, Union
  from fastNLP.envs.imports import _NEED_IMPORT_ONEFLOW
  from .oneflow_driver import OneflowDriver
  from fastNLP.core.drivers.oneflow_driver.utils import replace_sampler, replace_batch_sampler
  from fastNLP.core.samplers import ReproducibleBatchSampler, ReproducibleSampler, re_instantiate_sampler, ReproduceBatchSampler
  from fastNLP.core.samplers import RandomSampler
  from fastNLP.core.log import logger
label_desc: Use image node_id 2 for calling the OneflowSingleDriver obj's underlying member method code with example usage: obj.setup() without return types