74796ff02e9425ca336f595978fe6e7c422c0378,tensorlayer/layers.py,,get_variables_with_name,#Any#Any#Any#,149

Before Change

    >>> dense_vars = get_variable_with_name("dense", True, True)
    
    print("  Get variables with %s" % name)
    t_vars = tf.trainable_variables() if train_only else tf.all_variables()
    d_vars = [var for var in t_vars if name in var.name]
    if printable:
        for idx, v in enumerate(d_vars):
            print("  got {:3}: {:15}   {}".format(idx, v.name, str(v.get_shape())))

After Change

    
    print("  [*] geting variables with %s" % name)
    // tvar = tf.trainable_variables() if train_only else tf.all_variables()
    if train_only:
        t_vars = tf.trainable_variables()
    else:
        try: // TF1.0
            t_vars = tf.global_variables()
        except: // TF0.12
            t_vars = tf.all_variables()

    d_vars = [var for var in t_vars if name in var.name]
    if printable:
        for idx, v in enumerate(d_vars):
            print("  got {:3}: {:15}   {}".format(idx, v.name, str(v.get_shape())))

Instances