FluxML / MLJFlux.jl

Wrapping deep learning models from the package Flux.jl for use in the MLJ.jl toolbox
http://fluxml.ai/MLJFlux.jl/
MIT License
145 stars 17 forks source link

Benchmarks some raw Flux / MLJFlux comparisons #167

Open ablaom opened 3 years ago

ablaom commented 3 years ago

A discourse user has already provided some code that might expedite such a comparison. It would be good to see if we can reproduce the slowdown on some data (eg, synthetic) and diagnose the problem.

ablaom commented 3 years ago

@ayush-1506 If you have some time, maybe you could investigate??

ayush-1506 commented 3 years ago

Self-assigning.

ablaom commented 3 years ago

@ayush-1506 FYI The discourse post has been updated and this may be helpful.