昇腾社区首页
中文
注册

set_if_load

功能描述

设置是否载入已训练过的模型。

函数原型

def set_if_load(if_load)

参数说明

参数名

类型

可选/必选

说明

if_load

bool

必选

是否加载已训练的模型。

  • True:加载已训练的模型。
  • False:不加载已训练的模型。

返回值说明

无返回值

使用示例

from mx_rec.util.initialize import set_if_load
set_if_load(True)