75cb8084111e035cd23e0aa3e777a0798bc39cf2,tensornets/utils.py,,save,#,155

Before Change

def save(model, weights_path):
    sess = tf.get_default_session()
    assert sess is not None, "The default session should be given."

    weights = get_weights(model)
    names = [w.name for w in weights]
    values = sess.run(weights)

After Change

def save(model, weights_path, sess):
    if sess is None:
        sess = tf.get_default_session()
        assert sess is not None, "The default session should be given."

    weights = get_weights(model)
    names = [w.name for w in weights]
    values = sess.run(weights)
    np.savez(weights_path, names=names, values=values)

Instances