import tensorflow as tf

                        help='data_location')
    args = parser.parse_args()

    vdata = np.load(args.data_location)

    tf.reset_default_graph()
    idim = (36, 64)
    keep_prob = tf.placeholder(tf.float32, name='keep_prob')
    tftrain = tf.placeholder(tf.bool, name='tftrain')
    batch_size=100
    if (args.experiment_type == "reach") or (args.experiment_type == "push"):
        idim = (48, 48)
    tfinput = tf.placeholder(tf.float32, (3, batch_size) + idim + (3, ), name='x')
    if args.experiment_type == "reach":
