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

Baseline for the Spoofing-aware Speaker Verification Challenge 2022

CV backbones including GhostNet, TinyNet and TNT, developed by Huawei Noah's Ark Lab.

Node-level Graph Regression with Deep Gaussian Process Models

The missing CMake project initializer

Code and dataset for ACL2018 paper "Exploiting Document Knowledge for Aspect-level Sentiment Classification"

ALBERT-pytorch-implementation - ALBERT pytorch implementation

Pytorch implementation of Value Iteration Networks (NIPS 2016 best paper)

Redash reset for python

Implementation of Neural Distance Embeddings for Biological Sequences (NeuroSEED) in PyTorch

This repository is the official implementation of Unleashing the Power of Contrastive Self-Supervised Visual Models via Contrast-Regularized Fine-Tuning (NeurIPS21).

"Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementation

Lama-cleaner: Image inpainting tool powered by LaMa

SSD-based Object Detection in PyTorch

Galactic and gravitational dynamics in Python

Detector for Log4Shell exploitation attempts

Official PyTorch Implementation of Learning Self-Similarity in Space and Time as Generalized Motion for Video Action Recognition, ICCV 2021

PECOS - Prediction for Enormous and Correlated Spaces

This is the source code of the 1st place solution for segmentation task (with Dice 90.32%) in 2021 CCF BDCI challenge.

OpenMMLab Computer Vision Foundation

[CVPR 2021] Rethinking Text Segmentation: A Novel Dataset and A Text-Specific Refinement Approach