search

JuliaDecisionFocusedLearning / InferOpt.jl

Combinatorial optimization layers for machine learning pipelines
https://juliadecisionfocusedlearning.github.io/InferOpt.jl/
MIT License
113 stars 4 forks source link

Extend `DifferentiableFrankWolfe` beyond vectors #17

Closed gdalle closed 2 years ago

gdalle commented 2 years ago

At the moment, DifferentiableFrankWolfe (see the giom branch) only works when θ is a vector, not a higher-dimensional array. It should be easy to implement flattening / unflattening to generalize it.

gdalle commented 2 years ago

Fixed by https://github.com/gdalle/ImplicitDifferentiation.jl/releases/tag/v0.2.0, just need to copy the code