def get_initializer(name):
if name == 'identity':
return identity_initializer()
elif name == 'orthogonal':
return tf.orthogonal_initializer(gain=1.0)
elif name == 'truncated_normal':
return tf.truncated_normal_initializer()
评论列表
文章目录