Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
2
221cb0332
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Wiki
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Package registry
Model registry
Operate
Environments
Terraform modules
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
GitLab community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
Summer2022
221cb0332
Commits
fd022b93
Commit
fd022b93
authored
Feb 15, 2022
by
zhouneng2
Browse files
Options
Downloads
Patches
Plain Diff
[FasterRCNN]fix import error
parent
8df874f4
No related branches found
No related tags found
No related merge requests found
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
official/cv/faster_rcnn/eval.py
+1
-0
1 addition, 0 deletions
official/cv/faster_rcnn/eval.py
official/cv/faster_rcnn/src/network_define.py
+3
-3
3 additions, 3 deletions
official/cv/faster_rcnn/src/network_define.py
with
4 additions
and
3 deletions
official/cv/faster_rcnn/eval.py
+
1
−
0
View file @
fd022b93
...
@@ -20,6 +20,7 @@ from collections import defaultdict
...
@@ -20,6 +20,7 @@ from collections import defaultdict
import
numpy
as
np
import
numpy
as
np
from
pycocotools.coco
import
COCO
from
pycocotools.coco
import
COCO
import
mindspore
as
ms
from
mindspore.common
import
set_seed
,
Parameter
from
mindspore.common
import
set_seed
,
Parameter
from
src.dataset
import
data_to_mindrecord_byte_image
,
create_fasterrcnn_dataset
,
parse_json_annos_from_txt
from
src.dataset
import
data_to_mindrecord_byte_image
,
create_fasterrcnn_dataset
,
parse_json_annos_from_txt
...
...
This diff is collapsed.
Click to expand it.
official/cv/faster_rcnn/src/network_define.py
+
3
−
3
View file @
fd022b93
...
@@ -15,10 +15,10 @@
...
@@ -15,10 +15,10 @@
"""
FasterRcnn training network wrapper.
"""
"""
FasterRcnn training network wrapper.
"""
import
time
import
time
import
numpy
as
np
import
mindspore.common.dtype
as
mstype
import
mindspore.ops
as
ops
import
mindspore.ops
as
ops
import
mindspore.nn
as
nn
import
mindspore.nn
as
nn
from
mindspore
import
ParameterTuple
from
mindspore
import
ParameterTuple
,
Tensor
from
mindspore.train.callback
import
Callback
from
mindspore.train.callback
import
Callback
from
mindspore.nn.wrap.grad_reducer
import
DistributedGradReducer
from
mindspore.nn.wrap.grad_reducer
import
DistributedGradReducer
...
@@ -134,7 +134,7 @@ class TrainOneStepCell(nn.Cell):
...
@@ -134,7 +134,7 @@ class TrainOneStepCell(nn.Cell):
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
ops
.
GradOperation
(
get_by_list
=
True
,
self
.
grad
=
ops
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
sens
=
ms
.
numpy
.
ones
((
1
,)
*
sens
).
a
stype
(
np
.
float32
)
self
.
sens
=
Tensor
([
sens
,],
m
stype
.
float32
)
self
.
reduce_flag
=
reduce_flag
self
.
reduce_flag
=
reduce_flag
if
reduce_flag
:
if
reduce_flag
:
self
.
grad_reducer
=
DistributedGradReducer
(
optimizer
.
parameters
,
mean
,
degree
)
self
.
grad_reducer
=
DistributedGradReducer
(
optimizer
.
parameters
,
mean
,
degree
)
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment