擅长:python、mysql、java
<p>我想知道使用<code>tf.contrib.distribute</code>来指定设备放置策略是否有效。</p>
<pre><code>def main(unused_argv):
"""Code to load training folds data pickle or generate one if not present"""
strategy = tf.contrib.distribute.OneDeviceStrategy(device='/gpu:0')
config = tf.estimator.RunConfig(train_distribute=strategy)
# Create the Estimator
mnist_classifier = tf.estimator.Estimator(
model_fn=cnn_model_fn2,
config=config,
model_dir="F:/python_machine_learning_codes/tmp/custom_age_adience_1")
......
</code></pre>