search

FluxML / FluxML-Community-Call-Minutes

The FluxML Community Team repo
50 stars 4 forks source link

Add vision models #2

Closed darsnack closed 2 years ago

darsnack commented 4 years ago

Create standard models for vision tasks.

Reference issue: darsnack/FluxModels.jl#1

darsnack commented 4 years ago

If we go with https://github.com/JuliaCommunity/ML-Coordination-Tracker/issues/3#issuecomment-642130082 then it seems like many of the other models will reside in separate packages. So maybe we can do the same for the vision models.

darsnack commented 3 years ago

Current progress:

Accepting help to finish this up.

cyborg1995 commented 3 years ago

Hello

I would love to work on the squeezenet model. I'm new to julia and to github but I'm familiar with dl using PyTorch. Could someone guide me on how to start working on this? Also, which repository to fork?

Thanks

darsnack commented 3 years ago

Hi @cyborg1995. The best way to help with this issue is to push https://github.com/FluxML/Metalhead.jl/pull/70 across the finish line. Most of the models are complete, but we need to train them and add the BSON weights. I will post more tomorrow on how to add the weights to the repo, but for now, it would be a big help if you can clone the fork and attempt to train each model to full accuracy. I think this would be a great way to help out and get familiar with Flux+Julia.

darsnack commented 2 years ago

Closing this since Metalhead has been refactored to address this concern.