hub.config

在PaddleHub中,RunConfig代表了在对Task进行Fine-tune时的运行配置。包括运行的epoch次数、batch的大小、是否使用GPU训练等。

Class hub.finetune.config.RunConfig

  1. hub.RunConfig(
  2. log_interval=10,
  3. eval_interval=100,
  4. use_pyreader=True,
  5. use_data_parallel=True,
  6. save_ckpt_interval=None,
  7. use_cuda=False,
  8. checkpoint_dir=None,
  9. num_epoch=10,
  10. batch_size=None,
  11. enable_memory_optim=False,
  12. strategy=None)`

参数:

  • log_interval: 打印训练日志的周期,默认为10。

  • eval_interval: 进行评估的周期,默认为100。

  • use_pyreader: 是否使用pyreader,默认True。

  • use_data_parallel: 是否使用并行计算,默认True。打开该功能依赖nccl库。

  • save_ckpt_interval: 保存checkpoint的周期,默认为None。

  • use_cuda: 是否使用GPU训练和评估,默认为False。

  • checkpoint_dir: checkpoint的保存目录,默认为None,此时会在工作目录下根据时间戳生成一个临时目录。

  • num_epoch: 运行的epoch次数,默认为10。

  • batch_size: batch大小,默认为None。

  • enable_memory_optim: 是否进行内存优化,默认为False。

  • strategy: finetune的策略。默认为None,此时会使用DefaultFinetuneStrategy策略。

返回

RunConfig

示例

  1. import paddlehub as hub
  2. config = hub.RunConfig(
  3. use_cuda=True,
  4. num_epoch=10,
  5. batch_size=32)

log_interval

获取RunConfig设置的log_interval属性

示例

  1. import paddlehub as hub
  2. config = hub.RunConfig()
  3. log_interval = config.log_interval()

eval_interval

获取RunConfig设置的eval_interval属性

示例

  1. import paddlehub as hub
  2. config = hub.RunConfig()
  3. eval_interval = config.eval_interval()

use_pyreader

获取RunConfig设置的use_pyreader属性

示例

  1. import paddlehub as hub
  2. config = hub.RunConfig()
  3. use_pyreader = config.use_pyreader()

use_data_parallel

获取RunConfig设置的use_data_parallel属性

示例

  1. import paddlehub as hub
  2. config = hub.RunConfig()
  3. use_data_parallel = config.use_data_parallel()

save_ckpt_interval

获取RunConfig设置的save_ckpt_interval属性

示例

  1. import paddlehub as hub
  2. config = hub.RunConfig()
  3. save_ckpt_interval = config.save_ckpt_interval()

use_cuda

获取RunConfig设置的use_cuda属性

示例

  1. import paddlehub as hub
  2. config = hub.RunConfig()
  3. use_cuda = config.use_cuda()

checkpoint_dir

获取RunConfig设置的checkpoint_dir属性

示例

  1. import paddlehub as hub
  2. config = hub.RunConfig()
  3. checkpoint_dir = config.checkpoint_dir()

num_epoch

获取RunConfig设置的num_epoch属性

示例

  1. import paddlehub as hub
  2. config = hub.RunConfig()
  3. num_epoch = config.num_epoch()

batch_size

获取RunConfig设置的batch_size属性

示例

  1. import paddlehub as hub
  2. config = hub.RunConfig()
  3. batch_size = config.batch_size()

strategy

获取RunConfig设置的strategy属性

示例

  1. import paddlehub as hub
  2. config = hub.RunConfig()
  3. strategy = config.strategy()

enable_memory_optim

获取RunConfig设置的enable_memory_optim属性

示例

  1. import paddlehub as hub
  2. config = hub.RunConfig()
  3. enable_memory_optim = config.enable_memory_optim()