Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
MMCV
Commits
f3d5d979
Commit
f3d5d979
authored
Apr 07, 2019
by
Cao Yuhang
Browse files
fix memory order in log text
parent
1263c91e
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
5 deletions
+6
-5
mmcv/runner/hooks/logger/text.py
mmcv/runner/hooks/logger/text.py
+6
-5
No files found.
mmcv/runner/hooks/logger/text.py
View file @
f3d5d979
...
@@ -36,6 +36,12 @@ class TextLoggerHook(LoggerHook):
...
@@ -36,6 +36,12 @@ class TextLoggerHook(LoggerHook):
log_str
+=
(
log_str
+=
(
'time: {log[time]:.3f}, data_time: {log[data_time]:.3f}, '
.
'time: {log[time]:.3f}, data_time: {log[data_time]:.3f}, '
.
format
(
log
=
runner
.
log_buffer
.
output
))
format
(
log
=
runner
.
log_buffer
.
output
))
# statistic memory
if
runner
.
mode
==
'train'
and
torch
.
cuda
.
is_available
():
mem
=
torch
.
cuda
.
max_memory_allocated
()
mem_mb
=
int
(
mem
/
(
1024
*
1024
))
mem_str
=
'memory: {}, '
.
format
(
mem_mb
)
log_str
+=
mem_str
log_items
=
[]
log_items
=
[]
for
name
,
val
in
runner
.
log_buffer
.
output
.
items
():
for
name
,
val
in
runner
.
log_buffer
.
output
.
items
():
if
name
in
[
'time'
,
'data_time'
]:
if
name
in
[
'time'
,
'data_time'
]:
...
@@ -43,10 +49,5 @@ class TextLoggerHook(LoggerHook):
...
@@ -43,10 +49,5 @@ class TextLoggerHook(LoggerHook):
if
isinstance
(
val
,
float
):
if
isinstance
(
val
,
float
):
val
=
'{:.4f}'
.
format
(
val
)
val
=
'{:.4f}'
.
format
(
val
)
log_items
.
append
(
'{}: {}'
.
format
(
name
,
val
))
log_items
.
append
(
'{}: {}'
.
format
(
name
,
val
))
# statistic memory
if
runner
.
mode
==
'train'
and
torch
.
cuda
.
is_available
():
mem
=
torch
.
cuda
.
max_memory_allocated
()
mem_mb
=
int
(
mem
/
(
1024
*
1024
))
log_items
.
append
(
'{}: {}'
.
format
(
'memory'
,
mem_mb
))
log_str
+=
', '
.
join
(
log_items
)
log_str
+=
', '
.
join
(
log_items
)
runner
.
logger
.
info
(
log_str
)
runner
.
logger
.
info
(
log_str
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment