Skip to content

Commit

Permalink
log dir rename
Browse files Browse the repository at this point in the history
  • Loading branch information
yinhaofeng committed Mar 26, 2021
1 parent 326fa51 commit eb621e7
Show file tree
Hide file tree
Showing 5 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion doc/visualization.md
Original file line number Diff line number Diff line change
Expand Up @@ -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面板,有一下两种方法供您选择:

Expand Down
2 changes: 1 addition & 1 deletion tools/infer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
2 changes: 1 addition & 1 deletion tools/static_infer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down
2 changes: 1 addition & 1 deletion tools/static_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion tools/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit eb621e7

Please sign in to comment.