增量模型保存与加载

使用介绍

如需在Rec SDK训练框架使用增量模型保存与加载功能,可在初始化框架时,配置init接口的以下参数:

示例代码

使用示例如下:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
from mx_rec.util.initialize import init
# set init
init(train_steps=args.train_steps,
     eval_steps=args.eval_steps,
      save_steps=args.save_checkpoints_steps,
      max_steps=args.max_steps,
      use_dynamic=use_dynamic,
      use_dynamic_expansion=use_dynamic_expansion,
      save_checkpoint_due_time=4,
      save_delta_checkpoints_secs=2,
      is_incremental_checkpoint=True,
      restore_model_version=3)