【发布时间】:2020-08-05 10:54:12
【问题描述】:
我目前正在使用定义如下的conv2D layer 训练convolutional neural network:
conv1 = tf.keras.layers.Conv2D(filters=64, kernel_size=(3,3), padding='SAME', activation='relu')(inputs)
我的理解是默认的 kernel_initializer 是 glorot_uniform,它的默认种子是“none”:
tf.keras.layers.Conv2D(
filters, kernel_size, strides=(1, 1), padding='valid', data_format=None,
dilation_rate=(1, 1), activation=None, use_bias=True,
kernel_initializer='glorot_uniform', bias_initializer='zeros',
kernel_regularizer=None, bias_regularizer=None, activity_regularizer=None,
kernel_constraint=None, bias_constraint=None, **kwargs
)
tf.compat.v1.keras.initializers.glorot_uniform(seed=None, dtype=tf.dtypes.float32)
我正在尝试生成可重现的代码,并且已经按照this StackOverflow post 设置了随机种子:
seed_num = 1
os.environ['PYTHONHASHSEED'] = '0'
np.random.seed(seed_num)
rn.seed(seed_num)
session_conf = tf.compat.v1.ConfigProto(intra_op_parallelism_threads=1, inter_op_parallelism_threads=1)
tf.random.set_seed(seed_num)
sess = tf.compat.v1.Session(graph=tf.compat.v1.get_default_graph(), config=session_conf)
K.set_session(sess)
glorot_uniform 使用的 tf.random.set_seed 种子编号是否在 conv2D layer 中?如果不是,在定义conv2D layer 时如何定义该种子?
【问题讨论】:
标签: python tensorflow keras random-seed reproducible-research