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

Source code for paper "ATP: AMRize Than Parse! Enhancing AMR Parsing with PseudoAMRs" @NAACL-2022

Code for "Multi-View Multi-Person 3D Pose Estimation with Plane Sweep Stereo"

IDRLnet, a Python toolbox for modeling and solving problems through Physics-Informed Neural Network (PINN) systematically.

Streaming over lightweight data transformations

Unofficial Implementation of RobustSTL: A Robust Seasonal-Trend Decomposition Algorithm for Long Time Series (AAAI 2019)

Addition of pseudotorsion caclulation eta, theta, eta', and theta' to barnaba package

pcnaDeep integrates cutting-edge detection techniques with tracking and cell cycle resolving models.

This is the code used in the paper "Entity Embeddings of Categorical Variables".

Understanding the Generalization Benefit of Model Invariance from a Data Perspective

A Repository of Community-Driven Natural Instructions

Semantically Contrastive Learning for Low-light Image Enhancement

Official pytorch implementation of the AAAI 2021 paper Semantic Grouping Network for Video Captioning

This is our ARTS test set, an enriched test set to probe Aspect Robustness of ABSA.

Robust Lane Detection via Expanded Self Attention (WACV 2022)

Neural Geometric Level of Detail: Real-time Rendering with Implicit 3D Shapes (CVPR 2021 Oral)

Python package provinding tools for artistic interactive applications using AI

People movement type classifier with YOLOv4 detection and SORT tracking.

code release for USENIX'22 paper `On the Security Risks of AutoML`

A minimal TPU compatible Jax implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis

Generic image compressor for machine learning. Pytorch code for our paper "Lossy compression for lossless prediction".