wassname/Clover-Edition
1
0
Fork 0
mirror of https://github.com/wassname/Clover-Edition.git synced 2026-09-09 11:13:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
nickwalton
2019-10-22 16:11:35 -06:00
parent ccc04408aa
commit e5d1ebafba
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/ctrl/training_utils/training.py
+5 -4
View File
@@ -78,7 +78,6 @@ def input_fn(params=None):
return train_data
# the dimension of the transformer
embedding_dim = 1280
@@ -166,7 +165,9 @@ run_config = tf.contrib.tpu.RunConfig(
input_partition_dims=[[1, 1], [1, 1]], per_host_input_for_training=3))
tf.logging.set_verbosity(tf.logging.INFO)
tf.global_variables_initializer()
estimator_model = tf.keras.estimator.model_to_estimator(keras_model=model, config=run_config)
model.fit(input_fn(), epochs=1)
estimator_model.train(input_fn=input_fn, steps=args.iterations)
# estimator_model = tf.keras.estimator.model_to_estimator(keras_model=model, config=run_config)
#
# estimator_model.train(input_fn=input_fn, steps=args.iterations)