如需在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) |