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
ModelZoo
SOLOv2-pytorch
Commits
fdb44305
Commit
fdb44305
authored
Oct 07, 2018
by
Kai Chen
Browse files
minor fix
parent
6b4613df
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
5 additions
and
5 deletions
+5
-5
tools/configs/r50_fpn_frcnn_1x.py
tools/configs/r50_fpn_frcnn_1x.py
+1
-1
tools/configs/r50_fpn_maskrcnn_1x.py
tools/configs/r50_fpn_maskrcnn_1x.py
+2
-2
tools/train.py
tools/train.py
+2
-2
No files found.
tools/configs/r50_fpn_frcnn_1x.py
View file @
fdb44305
...
...
@@ -131,7 +131,7 @@ lr_config = dict(
checkpoint_config
=
dict
(
interval
=
1
)
# yapf:disable
log_config
=
dict
(
interval
=
2
0
,
interval
=
5
0
,
hooks
=
[
dict
(
type
=
'TextLoggerHook'
),
# dict(type='TensorboardLoggerHook', log_dir=work_dir + '/log')
...
...
tools/configs/r50_fpn_maskrcnn_1x.py
View file @
fdb44305
...
...
@@ -144,10 +144,10 @@ lr_config = dict(
checkpoint_config
=
dict
(
interval
=
1
)
# yapf:disable
log_config
=
dict
(
interval
=
2
0
,
interval
=
5
0
,
hooks
=
[
dict
(
type
=
'TextLoggerHook'
),
#
(
'TensorboardLoggerHook',
dict(
log_dir=work_dir + '/log')
),
#
dict(type=
'TensorboardLoggerHook', log_dir=work_dir + '/log')
])
# yapf:enable
# runtime settings
...
...
tools/train.py
View file @
fdb44305
...
...
@@ -90,8 +90,8 @@ def main():
cfg
.
work_dir
=
args
.
work_dir
cfg
.
gpus
=
args
.
gpus
# add mmdet version to checkpoint as meta data
cfg
.
checkpoint_config
.
meta
=
dict
(
mmdet_version
=
__version__
)
cfg
.
checkpoint_config
.
config
=
cfg
.
text
cfg
.
checkpoint_config
.
meta
=
dict
(
mmdet_version
=
__version__
,
config
=
cfg
.
text
)
logger
=
get_logger
(
cfg
.
log_level
)
...
...
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