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

Segmentation in Style: Unsupervised Semantic Image Segmentation with Stylegan and CLIP

Interpolation-based reduced-order models

PyTorch implementations of the paper: "Learning Independent Instance Maps for Crowd Localization"

Pytorch implementation of PTNet for high-resolution and longitudinal infant MRI synthesis

SIMULEVAL A General Evaluation Toolkit for Simultaneous Translation

Dynamic Divide-and-Conquer Adversarial Training for Robust Semantic Segmentation （ICCV2021）

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

Intel® Nervana™ reference deep learning framework committed to best performance on all hardware

U-Net: Convolutional Networks for Biomedical Image Segmentation

Implementation of a Transformer using ReLA (Rectified Linear Attention)

Torch Implementation of "Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network"

Official implementation of "Generating 3D Molecules for Target Protein Binding"

TensorFlow code for the neural network presented in the paper: "Structural Language Models of Code" (ICML'2020)

Constructing Neural Network-Based Models for Simulating Dynamical Systems

Official pytorch implementation of paper "Inception Convolution with Efficient Dilation Search" (CVPR 2021 Oral).

(ImageNet pretrained models) The official pytorch implemention of the TPAMI paper "Res2Net: A New Multi-scale Backbone Architecture"

Code for Paper Predicting Osteoarthritis Progression via Unsupervised Adversarial Representation Learning

Machine learning evaluation metrics, implemented in Python, R, Haskell, and MATLAB / Octave

Corruption Invariant Learning for Re-identification

This repo contains the code for paper Inverse Weighted Survival Games