|
|
|
|
|
|
if steps != 0 and train_model: |
|
|
|
save_model(sess, model_path=model_path, steps=steps, saver=saver) |
|
|
|
env.close() |
|
|
|
export_graph(model_path, env_name) |
|
|
|
graph_name = (env_name.strip() |
|
|
|
.replace('.app', '').replace('.exe', '').replace('.x86_64', '').replace('.x86', '')) |
|
|
|
graph_name = os.path.basename(os.path.normpath(graph_name)) |
|
|
|
export_graph(model_path, graph_name) |