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

It is a simple library to speed up CLIP inference up to 3x (K80 GPU)

An implementation of DeepMind's Relational Recurrent Neural Networks in PyTorch.

Implementation of NÜWA, state of the art attention network for text to video synthesis, in Pytorch

Acoustic mosquito detection code with Bayesian Neural Networks

Learning to Predict Gradients for Semi-Supervised Continual Learning

Implements a fake news detection program using classifiers.

A program that can analyze videos according to the weights you select

Deep Learning Head Pose Estimation using PyTorch.

pytorch implementation for PointNet

A Strong Baseline for Image Semantic Segmentation

FS-Mol: A Few-Shot Learning Dataset of Molecules

Spatial-Temporal Transformer for Dynamic Scene Graph Generation, ICCV2021

PyTorch implementation of Deformable Convolution

BERT model training impelmentation using 1024 A100 GPUs for MLPerf Training v1.1

Personalized Transfer of User Preferences for Cross-domain Recommendation (PTUPCDR)

A user-friendly research and development tool built to standardize RL competency assessment for custom agents and environments.

K-PLUG: Knowledge-injected Pre-trained Language Model for Natural Language Understanding and Generation in E-Commerce (EMNLP Founding 2021)

Cycle Consistent Adversarial Domain Adaptation (CyCADA)

🌈 PyTorch Implementation for EMNLP'21 Findings "Reasoning Visual Dialog with Sparse Graph Learning and Knowledge Transfer"

App customer segmentation cohort rfm clustering