Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import argparse
- import tensorflow as tf
- args = argparse.ArgumentParser()
- args.add_argument("--dev_sample_percentage", type=float, default=.1)
- args.add_argument("--batch_size", type=int, default=64)
- config, unparsed = args.parse_known_args()
- ...
- with tf.Graph().as_default():
- session_conf = tf.ConfigProto(
- allow_soft_placement=config.allow_soft_placement,
- log_device_placement=config.log_device_placement)
Add Comment
Please, Sign In to add comment