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
79e40801
"vscode:/vscode.git/clone" did not exist on "380a0416249b085f7b58baab5fcdd4dc020b5137"
Commit
79e40801
authored
Aug 07, 2017
by
Eli Bixby
Browse files
Fix GPUConfig initialization
parent
1b0bf3e6
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
6 deletions
+9
-6
tutorials/image/cifar10_estimator/cifar10_main.py
tutorials/image/cifar10_estimator/cifar10_main.py
+9
-6
No files found.
tutorials/image/cifar10_estimator/cifar10_main.py
View file @
79e40801
...
@@ -497,12 +497,15 @@ def main(unused_argv):
...
@@ -497,12 +497,15 @@ def main(unused_argv):
eval_steps
=
num_eval_examples
//
FLAGS
.
eval_batch_size
eval_steps
=
num_eval_examples
//
FLAGS
.
eval_batch_size
# Session configuration.
# Session configuration.
sess_config
=
tf
.
ConfigProto
()
sess_config
=
tf
.
ConfigProto
(
sess_config
.
allow_soft_placement
=
True
allow_soft_placement
=
True
,
sess_config
.
log_device_placement
=
FLAGS
.
log_device_placement
log_device_placement
=
FLAGS
.
log_device_placement
,
sess_config
.
intra_op_parallelism_threads
=
FLAGS
.
num_intra_threads
intra_op_parallelism_threads
=
FLAGS
.
num_intra_threads
,
sess_config
.
inter_op_parallelism_threads
=
FLAGS
.
num_inter_threads
inter_op_parallelism_threads
=
FLAGS
.
num_inter_threads
,
sess_config
.
gpu_options
.
force_gpu_compatible
=
FLAGS
.
force_gpu_compatible
gpu_options
=
tf
.
GPUOptions
(
force_gpu_compatible
=
FLAGS
.
force_gpu_compatible
)
)
# Hooks that add extra logging that is useful to see the loss more often in
# Hooks that add extra logging that is useful to see the loss more often in
# the console as well as examples per second.
# the console as well as examples per second.
...
...
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