mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-09-19 12:10:28 +08:00
25 lines
820 B
Diff
25 lines
820 B
Diff
47c47
|
|
<
|
|
---
|
|
> import tensorflow as tf
|
|
228c228
|
|
< def _create_keras_model_fn(keras_model, custom_objects=None):
|
|
---
|
|
> def _create_keras_model_fn(keras_model, params=None, custom_objects=None):
|
|
239c239
|
|
< def model_fn(features, labels, mode):
|
|
---
|
|
> def model_fn(features, labels, mode, params=None):
|
|
448c448
|
|
< if keras_model._is_graph_network:
|
|
---
|
|
> if False:
|
|
462,464c462,464
|
|
< estimator = estimator_lib.Estimator(keras_model_fn,
|
|
< config=config,
|
|
< warm_start_from=warm_start_path)
|
|
---
|
|
> estimator = tf.contrib.tpu.TPUEstimator(keras_model_fn, use_tpu=False, train_batch_size=4, eval_batch_size=4,
|
|
> config=config,
|
|
> warm_start_from=warm_start_path)
|