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
53ff5d90
"vscode:/vscode.git/clone" did not exist on "3eb7cbbccdfffcefaa0da974520e2e4a6da67a03"
Commit
53ff5d90
authored
Dec 19, 2018
by
Shining Sun
Browse files
Resolve merge conflict
parents
26a2c3b9
42420ce8
Changes
2
Expand all
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
305 additions
and
4 deletions
+305
-4
official/resnet/keras/keras_cifar_main.py
official/resnet/keras/keras_cifar_main.py
+3
-4
official/resnet/keras/resnet56.py
official/resnet/keras/resnet56.py
+302
-0
No files found.
official/resnet/keras/keras_cifar_main.py
View file @
53ff5d90
...
...
@@ -28,7 +28,7 @@ import tensorflow as tf # pylint: disable=g-bad-import-order
from
official.resnet
import
cifar10_main
as
cifar_main
from
official.resnet
import
resnet_run_loop
from
official.resnet.keras
import
keras_common
from
official.resnet.keras
import
keras_resnet_model
from
official.resnet.keras
import
resnet56
from
official.utils.flags
import
core
as
flags_core
from
official.utils.logs
import
logger
from
official.utils.misc
import
distribution_utils
...
...
@@ -146,13 +146,12 @@ def run(flags_obj):
optimizer
=
keras_common
.
get_optimizer
()
strategy
=
keras_common
.
get_dist_strategy
()
model
=
keras_resnet_model
.
ResNet56
(
input_shape
=
(
32
,
32
,
3
),
classes
=
cifar_main
.
_NUM_CLASSES
)
model
=
resnet56
.
ResNet56
(
input_shape
=
(
32
,
32
,
3
),
classes
=
cifar_main
.
_NUM_CLASSES
)
model
.
compile
(
loss
=
'categorical_crossentropy'
,
optimizer
=
optimizer
,
metrics
=
[
'categorical_accuracy'
],
distribute
=
strategy
)
time_callback
,
tensorboard_callback
,
lr_callback
=
keras_common
.
get_fit_callbacks
(
learning_rate_schedule
)
...
...
official/resnet/keras/
keras_resnet_model
.py
→
official/resnet/keras/
resnet56
.py
View file @
53ff5d90
This diff is collapsed.
Click to expand it.
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