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
f24c44d5
"docs/vscode:/vscode.git/clone" did not exist on "b70e333ac5be307d2d5e38c1653b755907a09f04"
Commit
f24c44d5
authored
Jul 27, 2017
by
Marianne Linhares Monteiro
Committed by
GitHub
Jul 27, 2017
Browse files
Fixing learning_rate to consider all workers
parent
943dff84
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
tutorials/image/cifar10_estimator/cifar10_main.py
tutorials/image/cifar10_estimator/cifar10_main.py
+1
-1
No files found.
tutorials/image/cifar10_estimator/cifar10_main.py
View file @
f24c44d5
...
...
@@ -226,7 +226,7 @@ def _resnet_model_fn(features, labels, mode):
# https://github.com/ppwwyyxx/tensorpack/blob/master/examples/ResNet/cifar10-resnet.py#L155
# users could apply other scheduling.
num_batches_per_epoch
=
cifar10
.
Cifar10DataSet
.
num_examples_per_epoch
(
'train'
)
//
FLAGS
.
train_batch_size
'train'
)
//
(
FLAGS
.
train_batch_size
*
FLAGS
.
num_workers
)
boundaries
=
[
num_batches_per_epoch
*
x
for
x
in
np
.
array
([
82
,
123
,
300
],
dtype
=
np
.
int64
)
...
...
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