def save_params()

in utils.py [0:0]

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

def save_params(sess, path):
    if mpi_rank() == 0:
        tf_vars = dict(zip([var.name for var in get_variables()],
                           sess.run(get_variables())))
        np.savez(path + '.npz', **tf_vars)