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
ResNet50_tensorflow
Commits
a704cd45
Unverified
Commit
a704cd45
authored
Jul 24, 2018
by
pkulzc
Committed by
GitHub
Jul 24, 2018
Browse files
Update learning_schedules.py for object detection python3 compatibility
parents
7922c9eb
ef84dca1
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
research/object_detection/utils/learning_schedules.py
research/object_detection/utils/learning_schedules.py
+1
-1
No files found.
research/object_detection/utils/learning_schedules.py
View file @
a704cd45
...
...
@@ -169,7 +169,7 @@ def manual_stepping(global_step, boundaries, rates, warmup=False):
boundaries
=
[
0
]
+
boundaries
num_boundaries
=
len
(
boundaries
)
rate_index
=
tf
.
reduce_max
(
tf
.
where
(
tf
.
greater_equal
(
global_step
,
boundaries
),
range
(
num_boundaries
),
list
(
range
(
num_boundaries
)
)
,
[
0
]
*
num_boundaries
))
return
tf
.
reduce_sum
(
rates
*
tf
.
one_hot
(
rate_index
,
depth
=
num_boundaries
),
name
=
'learning_rate'
)
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