import tensorflow as tf

    '''
    Print all trainable and non-trainable variables
    '''
    if train_only:
        t_vars = tf.trainable_variables()
        print('[*] printing trainable variables')
    else:
        try:
