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 DeepDream algorithm

OOD Generalization and Detection (ACL 2020)

Using knowledge-informed machine learning on the PRONOSTIA (FEMTO) and IMS bearing data sets. Predict remaining-useful-life (RUL).

Learning to See by Looking at Noise

This repository contain code on Novelty-Driven Binary Particle Swarm Optimisation for Truss Optimisation Problems.

A collection of 100 Deep Learning images and visualizations

Revisiting Video Saliency: A Large-scale Benchmark and a New Model (CVPR18, PAMI19)

BlockUnexpectedPackets - Preventing BungeeCord CPU overload due to Layer 7 DDoS attacks by scanning BungeeCord's logs

Sky Computing: Accelerating Geo-distributed Computing in Federated Learning

Vit-ImageClassification - Pytorch ViT for Image classification on the CIFAR10 dataset

Neural Style and MSG-Net

Rethinking the U-Net architecture for multimodal biomedical image segmentation

Repository for "Space-Time Correspondence as a Contrastive Random Walk" (NeurIPS 2020)

PyTorch code for Composing Partial Differential Equations with Physics-Aware Neural Networks

To build a regression model to predict the concrete compressive strength based on the different features in the training data.

Official implementation for the paper: Multi-label Classification with Partial Annotations using Class-aware Selective Loss

Semantic graph parser based on Categorial grammars

competitions-v2

An LSTM for time-series classification

Keras community contributions