tune_mnist_ray.py 文件源码

python
阅读 28 收藏 0 点赞 0 评论 0

项目:ray 作者: ray-project 项目源码 文件源码
def train(config={'activation': 'relu'}, reporter=None):
    global FLAGS, status_reporter, activation_fn
    status_reporter = reporter
    activation_fn = getattr(tf.nn, config['activation'])
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '--data_dir', type=str, default='/tmp/tensorflow/mnist/input_data',
        help='Directory for storing input data')
    FLAGS, unparsed = parser.parse_known_args()
    tf.app.run(main=main, argv=[sys.argv[0]] + unparsed)


# !!! Example of using the ray.tune Python API !!!
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号