【发布时间】:2021-03-11 03:54:01
【问题描述】:
我正在尝试将 Caffe 模型转换为 tensorflow。在 Caffe 中,我的卷积初始化如下:
weight_filler {
type: "gaussian"
std: 0.01
}
bias_filler {
type: "constant"
value: 0
}
如何在 Tensorflow 中获得相同的初始化?
【问题讨论】:
标签: tensorflow caffe
我正在尝试将 Caffe 模型转换为 tensorflow。在 Caffe 中,我的卷积初始化如下:
weight_filler {
type: "gaussian"
std: 0.01
}
bias_filler {
type: "constant"
value: 0
}
如何在 Tensorflow 中获得相同的初始化?
【问题讨论】:
标签: tensorflow caffe
使用tf.truncated_normal,您可以从截断的随机分布中获取随机值,因此:
shape = [10,10] # variable shape
weight = tf.Variable(tf.truncated_normal(shape, mean=0.0, stddev=0.01))
bias = tf.Variable(0.)
【讨论】:
initializer=tf.truncated_normal_initializer(mean=0.0,stddev=0.01)。感谢您的帮助