def pack_weights(self, w1, b1, w2, b2, loggamma, loglambda):
    params = np.concatenate([w1.flatten(), b1, w2, [b2], [loggamma], [loglambda]])
    return params
def pack_weights(self, w1, b1, w2, b2, loggamma, loglambda):
    params = np.concatenate([w1.flatten(), b1, w2, [b2], [loggamma], [loglambda]])
    return params