69d9220f450db20ce2e43a7c7f095b4d891565cc,tensorlayer/utils.py,,set_gpu_fraction,#Any#,634

Before Change

    
    tl.logging.info("[TL]: GPU MEM Fraction %f" % gpu_fraction)
    gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=gpu_fraction)
    sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
    return sess

After Change

    
    tl.logging.info("[TL]: GPU MEM Fraction %f" % gpu_fraction)
    if gpu_fraction is None:
        tf.config.gpu.set_per_process_memory_growth()
    else:
        tf.config.gpu.set_per_process_memory_fraction(0.4)
    // gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=gpu_fraction)
    // sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
    // return sess


def train_epoch(network, X, y, cost, train_op=tf.optimizers.Adam(learning_rate=0.0001),
                acc=None, batch_size=100, shuffle=True):
    Training a given non time-series network by the given cost function, training data, batch_size etc.
    for one epoch.

Instances