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
26ee1a2f
Unverified
Commit
26ee1a2f
authored
Nov 30, 2018
by
John Arevalo
Committed by
GitHub
Nov 30, 2018
Browse files
Fix #5814
Comments wrongly say dropout is used
parent
5324fc66
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
3 deletions
+2
-3
samples/core/get_started/custom_estimator.py
samples/core/get_started/custom_estimator.py
+2
-3
No files found.
samples/core/get_started/custom_estimator.py
View file @
26ee1a2f
...
...
@@ -27,9 +27,8 @@ parser.add_argument('--train_steps', default=1000, type=int,
help
=
'number of training steps'
)
def
my_model
(
features
,
labels
,
mode
,
params
):
"""DNN with three hidden layers, and dropout of 0.1 probability."""
# Create three fully connected layers each layer having a dropout
# probability of 0.1.
"""DNN with three hidden layers and learning_rate=0.1."""
# Create three fully connected layers.
net
=
tf
.
feature_column
.
input_layer
(
features
,
params
[
'feature_columns'
])
for
units
in
params
[
'hidden_units'
]:
net
=
tf
.
layers
.
dense
(
net
,
units
=
units
,
activation
=
tf
.
nn
.
relu
)
...
...
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