def get_tf_params()

in understanding_rl_vision/rl_clarity/training.py [0:0]

• 12 lines of code
• 5 McCabe index (conditional complexity)

def get_tf_params(scope):
    """
    Get a dictionary of parameters from TensorFlow for the specified scope
    """
    import tensorflow as tf
    from baselines.common.tf_util import get_session

    sess = get_session()
    allvars = tf.trainable_variables(scope)
    nonopt_vars = [
        v
        for v in allvars
        if all(veto not in v.name for veto in ["optimizer", "kbuf", "vbuf"])
    ]
    name2var = {v.name: v for v in nonopt_vars}
    return sess.run(name2var)