Skip to content

G
gm_strategy_cvr
Commit ed7d8cf8 authored by 赵威's avatar 赵威
Browse files
Options

update config

parent 4e76bdee
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
src/main.py
View file @ ed7d8cf8
...@@ -40,10 +40,10 @@ def main(): ...@@ -40,10 +40,10 @@ def main():
all_features = build_features(df) all_features = build_features(df)
params = {"feature_columns": all_features, "hidden_units": [64, 32], "learning_rate": 0.1} params = {"feature_columns": all_features, "hidden_units": [64, 32], "learning_rate": 0.1}
model_path = str(Path("~/data/model_tmp/").expanduser()) model_path = str(Path("~/data/model_tmp/").expanduser())
# if os.path.exists(model_path): if os.path.exists(model_path):
# shutil.rmtree(model_path) shutil.rmtree(model_path)
session_config = tf.ConfigProto() session_config = tf.compat.v1.ConfigProto()
session_config.gpu_options.allow_growth = True session_config.gpu_options.allow_growth = True
session_config.gpu_options.per_process_gpu_memory_fraction = 0.9 session_config.gpu_options.per_process_gpu_memory_fraction = 0.9
estimator_config = tf.estimator.RunConfig(session_config=session_config) estimator_config = tf.estimator.RunConfig(session_config=session_config)
......
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