Allow two_phase_solve to have tree constraints

gehring / fax

MIT License

78 stars 9 forks source link

Allow two_phase_solve to have tree constraints #33

Closed evgenii-nikishin closed 4 years ago

evgenii-nikishin commented 4 years ago

Right now, two_phase_solve assumes that param_func's output and init_xs argument are vector-valued. Could you please allow them to be represented as trees (e.g. if init_xs is NN params)?

gehring commented 4 years ago

Resolved in #34