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
2d34cebe
Commit
2d34cebe
authored
Oct 01, 2018
by
pangjm
Browse files
validate api transfer to hook
parent
c20c2c88
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
10 deletions
+5
-10
tools/train.py
tools/train.py
+5
-10
No files found.
tools/train.py
View file @
2d34cebe
...
@@ -134,12 +134,6 @@ def main():
...
@@ -134,12 +134,6 @@ def main():
runner
=
Runner
(
model
,
batch_processor
,
cfg
.
optimizer
,
cfg
.
work_dir
,
runner
=
Runner
(
model
,
batch_processor
,
cfg
.
optimizer
,
cfg
.
work_dir
,
cfg
.
log_level
)
cfg
.
log_level
)
if
args
.
validate
:
val_dataset
=
obj_from_dict
(
cfg
.
data
.
test
,
datasets
)
data_loaders
.
append
(
build_dataloader
(
val_dataset
,
cfg
.
data
.
imgs_per_gpu
,
cfg
.
data
.
workers_per_gpu
,
cfg
.
gpus
,
dist
))
# register hooks
# register hooks
optimizer_config
=
DistOptimizerHook
(
optimizer_config
=
DistOptimizerHook
(
**
cfg
.
optimizer_config
)
if
dist
else
cfg
.
optimizer_config
**
cfg
.
optimizer_config
)
if
dist
else
cfg
.
optimizer_config
...
@@ -148,10 +142,11 @@ def main():
...
@@ -148,10 +142,11 @@ def main():
if
dist
:
if
dist
:
runner
.
register_hook
(
DistSamplerSeedHook
())
runner
.
register_hook
(
DistSamplerSeedHook
())
# register eval hooks
# register eval hooks
if
isinstance
(
model
.
module
,
RPN
):
if
args
.
validate
:
runner
.
register_hook
(
CocoDistEvalRecallHook
(
cfg
.
data
.
val
))
if
isinstance
(
model
.
module
,
RPN
):
elif
cfg
.
data
.
val
.
type
==
'CocoDataset'
:
runner
.
register_hook
(
CocoDistEvalRecallHook
(
cfg
.
data
.
val
))
runner
.
register_hook
(
CocoDistEvalmAPHook
(
cfg
.
data
.
val
))
elif
cfg
.
data
.
val
.
type
==
'CocoDataset'
:
runner
.
register_hook
(
CocoDistEvalmAPHook
(
cfg
.
data
.
val
))
if
cfg
.
resume_from
:
if
cfg
.
resume_from
:
runner
.
resume
(
cfg
.
resume_from
)
runner
.
resume
(
cfg
.
resume_from
)
...
...
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