tensorflow估计器:过滤模型的初始化值

时间:2018-11-09 12:07:29

标签: python tensorflow tensorflow-estimator

我正在使用tensorflow估计器api并成功地训练了我的模型。但是,我想看看训练开始时如何初始化我的过滤器。

此刻,我可以使用以下方法强制模型从某个训练阶段为我提供内核值:

print(tf.get_default_graph().get_tensor_by_name('resnet_model/conv2d/kernel:0'))

当我用估计量进行预测时

estimator.predict(input_fn=input_fn,checkpoint_path = checkpointfile)

但是我也想在网络学到任何东西之前获取初始化值。所以我的想法是不给模型任何检查点。这会导致错误:

ValueError: Could not find trained model in model_dir

那么在使用estimator API时如何获取过滤器的初始化值?

0 个答案:

没有答案