Open
Description
Hi everyone, @svekars and I are looking to increase the number of new contributions to pytorch/examples, this might be especially interesting to you if you've never contributed to an open source project before.
At a high level, we're looking for new interesting models.
So here's what you need to do
- Check out our contributing guide: https://github.com/pytorch/examples/blob/main/CONTRIBUTING.md
- Pick a model idea - I've listed a few below, comment on this task so others know you're working on it
- Implement your model from scratch using PyTorch, no external dependencies will be allowed to keep the examples as educational as possible
Your implementation needs to include
- A folder with your code which needs to define
- Your model architecture
- Training code
- Evaluation code
- An argparser
- Make sure your script runs in CI so it doesn't break in the future by adding it to
run_python_examples.sh
- README describing any usage instructions
As an example this recent contribution by @sudomaze is a good one to follow https://github.com/pytorch/examples/pull/1003/files
Here are some model ideas
Model ideas
- Controlnet - Guided diffusion
- NERF
- Graph Neural Network @JoseLuisC99
- Diffusion Model, stable diffusion or any variant of the architecture you like
- Vision Transformer
- Video model
- Toolformer
- Differentiable physics
- Flownet
- Dreamfusion or any 3d model
- Language Translation
- Swin transformer
But I'm quite open to anything we don't have that's cool