Skip to content

S
serviceRec
Commit 04b1805f authored by 宋柯's avatar 宋柯
Browse files
Options

模型调试

parent 0713f4c7
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
train/train_service_sk.py
View file @ 04b1805f
...@@ -214,7 +214,7 @@ session_config.gpu_options.allow_growth = True ...@@ -214,7 +214,7 @@ session_config.gpu_options.allow_growth = True
# config = tf.estimator.RunConfig(save_checkpoints_steps = 10000, train_distribute = distribution, eval_distribute = distribution) # config = tf.estimator.RunConfig(save_checkpoints_steps = 10000, train_distribute = distribution, eval_distribute = distribution)
config = tf.estimator.RunConfig(save_checkpoints_steps = 10000, session_config = session_config) config = tf.estimator.RunConfig(save_checkpoints_steps = 3000, session_config = session_config)
wideAndDeepModel = tf.estimator.DNNLinearCombinedClassifier(model_dir = BASE_DIR + 'model', wideAndDeepModel = tf.estimator.DNNLinearCombinedClassifier(model_dir = BASE_DIR + 'model',
linear_feature_columns = linear_feature_columns, linear_feature_columns = linear_feature_columns,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment