added_training_stuff

2026-09-19 12:10:28 +08:00 · 2019-09-27 14:39:18 -06:00
parent ef45ad99d0
commit d71a2e95eb
14 changed files with 447998 additions and 1 deletions
@@ -0,0 +1,24 @@
+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)