-
Notifications
You must be signed in to change notification settings - Fork 1.2k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
add visualdl #137
add visualdl #137
Conversation
tools/program.py
Outdated
@@ -412,6 +415,10 @@ def run(dataloader, exe, program, fetchs, epoch=0, mode='train'): | |||
metric_list[i].update(m[0], len(batch[0])) | |||
fetchs_str = ''.join([str(m.value) + ' ' | |||
for m in metric_list] + [batch_time.value]) + 's' | |||
if vdl_dir: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
if vdl_dir -> if vdl_dir is not None ?
tools/train.py
Outdated
parser.add_argument( | ||
'--vdl_dir', | ||
type=str, | ||
default="scaler", |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
default as None?
@@ -38,6 +39,11 @@ def parse_args(): | |||
type=str, | |||
default='configs/ResNet/ResNet50.yaml', | |||
help='config file path') | |||
parser.add_argument( | |||
'--vdl_dir', |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
文档那块儿 是不是也得加一下这个args的
add optimize test
This PR integrates VisualDL into PaddleClas
One can monitor statistics(loss, acc) in real time