def session_config():
optimizer_options = tf.OptimizerOptions(opt_level=tf.OptimizerOptions.L0)
config = tf.ConfigProto(
graph_options=tf.GraphOptions(optimizer_options=optimizer_options))
config.operation_timeout_in_ms = 10*1000 # abort after 10 seconds
return config
评论列表
文章目录