通过scope初始化batch norm的可训练参数
Created by: wanghaoshuang
获取所有parameters:
for var in fluid.default_main_program().list_vars():
if var.persistable():
print var.name, var.shape
观察确定batch_norm op相关的四个参数的名称。 直接通过scope初始化指定变量:
place = fluid.CPUPlace()
param_t = fluid.global_scope().find_var(var_name).get_tensor()
param_t.set(np.ndarray([5, 30]), place)