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
3d166f95
"tests/vscode:/vscode.git/clone" did not exist on "72d3a30c6327e70de3595d00f04e2d577fcbbb68"
Commit
3d166f95
authored
Feb 08, 2018
by
Neal Wu
Browse files
Swap tf.constant with tf.random_uniform for consistency
parent
c20eb5ff
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
4 deletions
+2
-4
official/resnet/cifar10_test.py
official/resnet/cifar10_test.py
+1
-2
official/resnet/imagenet_test.py
official/resnet/imagenet_test.py
+1
-2
No files found.
official/resnet/cifar10_test.py
View file @
3d166f95
...
...
@@ -113,8 +113,7 @@ class BaseTest(tf.test.TestCase):
model
=
cifar10_main
.
Cifar10Model
(
32
,
data_format
=
'channels_last'
,
num_classes
=
num_classes
)
fake_input
=
tf
.
constant
(
0.0
,
shape
=
[
batch_size
,
_HEIGHT
,
_WIDTH
,
_NUM_CHANNELS
])
fake_input
=
tf
.
random_uniform
([
batch_size
,
_HEIGHT
,
_WIDTH
,
_NUM_CHANNELS
])
output
=
model
(
fake_input
,
training
=
True
)
self
.
assertAllEqual
(
output
.
shape
,
(
batch_size
,
num_classes
))
...
...
official/resnet/imagenet_test.py
View file @
3d166f95
...
...
@@ -182,8 +182,7 @@ class BaseTest(tf.test.TestCase):
model
=
imagenet_main
.
ImagenetModel
(
50
,
data_format
=
'channels_last'
,
num_classes
=
num_classes
)
fake_input
=
tf
.
constant
(
0.0
,
shape
=
[
batch_size
,
224
,
224
,
3
])
fake_input
=
tf
.
random_uniform
([
batch_size
,
224
,
224
,
3
])
output
=
model
(
fake_input
,
training
=
True
)
self
.
assertAllEqual
(
output
.
shape
,
(
batch_size
,
num_classes
))
...
...
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