wassname/Clover-Edition
1
0 Fork 0
mirror of https://github.com/wassname/Clover-Edition.git synced 2026-10-29 12:40:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
nickwalton committed 2019-10-22 16:15:02 -06:00
1 parent 9939bea3f8
commit 6510c44ba1
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
generator/ctrl/training_utils/training.py
+1 -1
View File
@@ -166,7 +166,7 @@ run_config = tf.contrib.tpu.RunConfig(
tf.logging.set_verbosity(tf.logging.INFO)
params = {"batch_size": 2}
model.fit(input_fn(params=params), epochs=1)
model.fit(input_fn(params=params), steps_per_epoch=1000, epochs=1)
# estimator_model = tf.keras.estimator.model_to_estimator(keras_model=model, config=run_config)