pytorch-lightning-conference-seed

by PyTorchLightning

Pytorch Lightning code guideline for conferences

330 Stars 59 Forks Last release: Not found Apache License 2.0 91 Commits 0 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

Deep learning project seed

Use this seed to start new deep learning / ML projects.

  • Built in setup.py
  • Built in requirements
  • Examples with MNIST
  • Badges
  • Bibtex

Goals

The goal of this seed is to structure ML paper-code the same so that work can easily be extended and replicated.

DELETE EVERYTHING ABOVE FOR YOUR PROJECT


Your Project Name

Paper Conference Conference Conference

CI testing

Description

What it does

How to run

First, install dependencies
```bash

clone project

git clone https://github.com/YourGithubName/deep-learning-project-template

install project

cd deep-learning-project-template pip install -e .
pip install -r requirements.txt

 Next, navigate to any file and run it.   
bash

module folder

cd project

run module (example: mnist as your main contribution)

python litclassifiermain.py
```

Imports

This project is setup as a package which means you can now easily import any file into any other file like so: ```python from project.datasets.mnist import mnist from project.litclassifiermain import LitClassifier from pytorch_lightning import Trainer

model

model = LitClassifier()

data

train, val, test = mnist()

train

trainer = Trainer() trainer.fit(model, train, val)

test using the best model!

trainer.test(test_dataloaders=test) ```

Citation

@article{YourName,
  title={Your Title},
  author={Your team},
  journal={Location},
  year={Year}
}

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.