NPUBroadcastGlobalVariablesCallback构造函数

函数原型

def __init__(self, root_rank)

功能说明

Keras场景下对变量进行广播，使得在分布式场景下每个device上的变量初始值保持一致。

参数说明

返回值

无

调用示例

迁移前：

callbacks = [hvd.callbacks.BroadcastGlobalVariablesCallback(0)]

import numpy as np
data = np.random.random((1000, 100))
lables = np random.randint(2, size=(1000,1))
model.fit(data, lables, epochs=10, batch_size=32, callbacks=callbacks)

迁移后：

from npu_bridge.npu_init import *
callbacks = [NPUBroadcastGlobalVariablesCallback(0)]

import numpy as np
data = np.random.random((1000, 100))
lables = np random.randint(2, size=(1000,1))
model.fit(data, lables, epochs=10, batch_size=32, callbacks=callbacks)