search

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

Error `fit` GPU #247

Closed andrewrosemberg closed 9 months ago

andrewrosemberg commented 11 months ago

Sorry if this is not the place for this issue! I am not quite sure where to raise it

I am hitting the same error as the one in the CI build.

I understand this is an upstream issue. Do you know if this issue has been raised elsewhere?

ablaom commented 11 months ago

Thanks for flagging and looking into this!

I also notice the error (which has surfaced without any change to to the MLJFlux source code) but was still scratching my head about it.

I understand this is an upstream issue. Do you know if this issue has been raised elsewhere?

If you have isolated the issue, be great to note your progress here. No, I have not raised anything elsewhere.

andrewrosemberg commented 11 months ago

Understood. Yeah this is a weird bug. I am going through the functions of the Stacktrace to see which ones were modified upstream. If it is still not clear what the problem is, I will fix the packages upstream to older tags.

I will post here what I find.

andrewrosemberg commented 11 months ago

I believe the error is in Zygote.jl.

Other people have hit the error since a change there: https://github.com/FluxML/Zygote.jl/issues/1470

andrewrosemberg commented 9 months ago

Correct me if I am wrong, but I believe this is now fixed :) I will close the issue but feel free to re-open if I got it wrong.