"git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "a07007e5e16ea37fd6846c53d21dd4fb68b84675"
Commit 53ff5d90 authored by Shining Sun's avatar Shining Sun
Browse files

Resolve merge conflict

parents 26a2c3b9 42420ce8
...@@ -28,7 +28,7 @@ import tensorflow as tf # pylint: disable=g-bad-import-order ...@@ -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 cifar10_main as cifar_main
from official.resnet import resnet_run_loop from official.resnet import resnet_run_loop
from official.resnet.keras import keras_common 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.flags import core as flags_core
from official.utils.logs import logger from official.utils.logs import logger
from official.utils.misc import distribution_utils from official.utils.misc import distribution_utils
...@@ -146,13 +146,12 @@ def run(flags_obj): ...@@ -146,13 +146,12 @@ def run(flags_obj):
optimizer = keras_common.get_optimizer() optimizer = keras_common.get_optimizer()
strategy = keras_common.get_dist_strategy() strategy = keras_common.get_dist_strategy()
model = keras_resnet_model.ResNet56(input_shape=(32, 32, 3), model = resnet56.ResNet56(input_shape=(32, 32, 3),
classes=cifar_main._NUM_CLASSES) classes=cifar_main._NUM_CLASSES)
model.compile(loss='categorical_crossentropy', model.compile(loss='categorical_crossentropy',
optimizer=optimizer, optimizer=optimizer,
metrics=['categorical_accuracy'], metrics=['categorical_accuracy'],
distribute=strategy)
time_callback, tensorboard_callback, lr_callback = keras_common.get_fit_callbacks( time_callback, tensorboard_callback, lr_callback = keras_common.get_fit_callbacks(
learning_rate_schedule) learning_rate_schedule)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment