From eb621e7f2c73eda3e83baa6a3e06c7b338a74ba5 Mon Sep 17 00:00:00 2001 From: yinhaofeng <1841837261@qq.com> Date: Fri, 26 Mar 2021 07:21:31 +0000 Subject: [PATCH] log dir rename --- doc/visualization.md | 2 +- tools/infer.py | 2 +- tools/static_infer.py | 2 +- tools/static_trainer.py | 2 +- tools/trainer.py | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/doc/visualization.md b/doc/visualization.md index 611b9aecc..c3fa99740 100644 --- a/doc/visualization.md +++ b/doc/visualization.md @@ -9,7 +9,7 @@ python -m pip install visualdl -i https://mirror.baidu.com/pypi/simple ## 开启可视化功能 1. 在各模型的yaml配置文件中，runner项下添加新的参数“use_visual”，并将该项的值填写为True。该参数为bool类型，默认值为False，用于在安装VisualDL完成的情况下开启可视化训练。 -2. 在模型的dygraph_model.py文件中，可以通过train_forward函数的metrics_list, print_dict两个返回值来输出动态图运行时您需要打印的指标或变量。同理在模型的static_model.py文件中，可以通过net函数的fetch_dict返回值来输出静态图运行时您需要打印的指标。可视化功能会自动收集这些指标，并创建一个log目录存放他们。 +2. 在模型的dygraph_model.py文件中，可以通过train_forward函数的metrics_list, print_dict两个返回值来输出动态图运行时您需要打印的指标或变量。同理在模型的static_model.py文件中，可以通过net函数的fetch_dict返回值来输出静态图运行时您需要打印的指标。可视化功能会自动收集这些指标，并创建一个visualDL_log目录存放他们。 3. 您可以正常的训练模型 4. 启动VisualDL面板，有一下两种方法供您选择： diff --git a/tools/infer.py b/tools/infer.py index 4cb0a4428..b6d6cce67 100644 --- a/tools/infer.py +++ b/tools/infer.py @@ -85,7 +85,7 @@ def main(args): # Create a log_visual object and store the data in the path if use_visual: from visualdl import LogWriter - log_visual = LogWriter(args.abs_dir + "/log/infer") + log_visual = LogWriter(args.abs_dir + "/visualDL_log/infer") # to do : add optimizer function #optimizer = dy_model_class.create_optimizer(dy_model, config) diff --git a/tools/static_infer.py b/tools/static_infer.py index 165d349db..b16c05193 100644 --- a/tools/static_infer.py +++ b/tools/static_infer.py @@ -85,7 +85,7 @@ def main(args): # Create a log_visual object and store the data in the path if use_visual: from visualdl import LogWriter - log_visual = LogWriter(args.abs_dir + "/log/infer") + log_visual = LogWriter(args.abs_dir + "/visualDL_log/infer") step_num = 0 for epoch_id in range(start_epoch, end_epoch): diff --git a/tools/static_trainer.py b/tools/static_trainer.py index 26b4b7f9d..d5e760f49 100644 --- a/tools/static_trainer.py +++ b/tools/static_trainer.py @@ -89,7 +89,7 @@ def main(args): # Create a log_visual object and store the data in the path if use_visual: from visualdl import LogWriter - log_visual = LogWriter(args.abs_dir + "/log/train") + log_visual = LogWriter(args.abs_dir + "/visualDL_log/train") else: log_visual = None step_num = 0 diff --git a/tools/trainer.py b/tools/trainer.py index 5a8d2113a..fd6544ccd 100644 --- a/tools/trainer.py +++ b/tools/trainer.py @@ -86,7 +86,7 @@ def main(args): # Create a log_visual object and store the data in the path if use_visual: from visualdl import LogWriter - log_visual = LogWriter(args.abs_dir + "/log/train") + log_visual = LogWriter(args.abs_dir + "/visualDL_log/train") if model_init_path is not None: load_model(model_init_path, dy_model)