search

FluxML / model-zoo

Please do not feed the models
https://fluxml.ai/
Other
918 stars 333 forks source link

School work #248

Open tberky opened 4 years ago

tberky commented 4 years ago

Hello, i need to do some stuff in Julia/Flux for my school work and i cannot find something that actually works for me into start here. After days of trying to repair all kind of codes from modelZOO i dont know what to do.

I need to do simple MLP(something for like BMI calculation or something) and then MLP and CNN for MNIST classification and then simple GRU/LTSM.

Can somebody here help? I tried julia 1.4.2. with FLux 0.10.3. and stable julia with the same flux, other flux versions doesnt work for me either. I am using atom with Juno...

Is there some documentation for on which versions where the files created?

AdarshKumar712 commented 4 years ago

Hey @tomiczech7 , have you tried Julia v1.3 with Flux 0.10.3? Try that once and while using any of the pre-existing models on modelzoo, use the corresponding project.toml file to load the pre-requisite packages.

lorrp1 commented 4 years ago

im runnning julia 1.5.1 with everything else updated and it seems fine.

you need to be careful with CuArray, do not use add it otherwise it will downgrade many other pkg (use CUDA instead) this one should works if you replace CuArray with CUDA https://github.com/FluxML/model-zoo/blob/master/vision/mnist/mlp.jl