模型调试

3be3af30 · 宋柯 · aaa78260 · 3be3af30
Commit 3be3af30 authored Dec 28, 2021 by 宋柯
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

train_service_sk_tf2_distibute.py train/train_service_sk_tf2_distibute.py +1 -1

No files found.
--- a/train/train_service_sk_tf2_distibute.py
+++ b/train/train_service_sk_tf2_distibute.py
@@ -216,7 +216,7 @@ session_config = tf.compat.v1.ConfigProto(allow_soft_placement = True)

 session_config.gpu_options.allow_growth = True

-config = tf.estimator.RunConfig(save_checkpoints_steps = 1500, train_distribute = distribution, eval_distribute = distribution)
+config = tf.estimator.RunConfig(save_checkpoints_steps = 3000, train_distribute = distribution, eval_distribute = distribution)

 # config = tf.estimator.RunConfig(save_checkpoints_steps = 3000, session_config = session_config)