Skip to content

New examples requested #1131

Open
Open
@msaroufim

Description

@msaroufim

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

  1. Check out our contributing guide: https://github.com/pytorch/examples/blob/main/CONTRIBUTING.md
  2. Pick a model idea - I've listed a few below, comment on this task so others know you're working on it
  3. 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

  1. A folder with your code which needs to define
  2. Your model architecture
  3. Training code
  4. Evaluation code
  5. An argparser
  6. Make sure your script runs in CI so it doesn't break in the future by adding it to run_python_examples.sh
  7. 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

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions