Lava-DL, but with PyTorch-Lightning flavour

Last update: Oct 31, 2022

Related tags

Deep Learning lava-dl-lightning

Overview

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

Description

What it does

How to run

Requires Python 3.8+. First, install dependencies

# clone project   
git clone https://github.com/Barchid/lava-dl-lightning
cd lava-dl-lightning

# install lava
wget https://github.com/lava-nc/lava/releases/download/v0.2.0/lava-nc-0.2.0.tar.gz
pip install -U pip
pip install lava-nc-0.2.0.tar.gz

# install lava-dl
wget https://github.com/lava-nc/lava-dl/releases/download/v0.1.1/lava-dl-0.1.1.tar.gz
pip install lava-dl-0.1.1.tar.gz

# install pytorch-lightning
pip install pytorch-lightning torchmetrics

Next, navigate to any file and run it.

# module folder
cd project

# run module (example: mnist as your main contribution)   
python lit_classifier_main.py

Imports

This project is setup as a package which means you can now easily import any file into any other file like so:

from project.datasets.mnist import mnist
from project.lit_classifier_main 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}
}

Lava-DL, but with PyTorch-Lightning flavour

Related tags

Overview

Deep learning project seed

Goals

DELETE EVERYTHING ABOVE FOR YOUR PROJECT

Your Project Name

Description

How to run

Imports

Citation

Owner

Sami BARCHID

PyTorch implementation of "Contrast to Divide: self-supervised pre-training for learning with noisy labels"

Tutorial repo for an end-to-end Data Science project

A graph-to-sequence model for one-step retrosynthesis and reaction outcome prediction.

Memory Defense: More Robust Classificationvia a Memory-Masking Autoencoder

A Robust Non-IoU Alternative to Non-Maxima Suppression in Object Detection

Code for binary and multiclass model change active learning, with spectral truncation implementation.

A Small and Easy approach to the BraTS2020 dataset (2D Segmentation)

Recognize numbers from an (28 x 28) image using neural networks

Learning the Beauty in Songs: Neural Singing Voice Beautifier; ACL 2022 (Main conference); Official code

Light-SERNet: A lightweight fully convolutional neural network for speech emotion recognition

OpenGAN: Open-Set Recognition via Open Data Generation

A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.

ONNX-PackNet-SfM: Python scripts for performing monocular depth estimation using the PackNet-SfM model in ONNX

Pytorch implementation for "Adversarial Robustness under Long-Tailed Distribution" (CVPR 2021 Oral)

A Framework for Encrypted Machine Learning in TensorFlow

A parametric soroban written with CADQuery.

PyTorch code to run synthetic experiments.

Train an RL agent to execute natural language instructions in a 3D Environment (PyTorch)

Turning SymPy expressions into PyTorch modules.

Learning with Noisy Labels via Sparse Regularization, ICCV2021