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

RoFormer_pytorch

Reproduces ResNet-V3 with pytorch

[NeurIPS-2020] Self-paced Contrastive Learning with Hybrid Memory for Domain Adaptive Object Re-ID.

Chatbot in 200 lines of code using TensorLayer

A copy of Ares that costs 30 fucking dollars.

Optical Character Recognition + Instance Segmentation for russian and english languages

This is the official repository of the paper Stocastic bandits with groups of similar arms (NeurIPS 2021). It contains the code that was used to compute the figures and experiments of the paper.

python library for invisible image watermark (blind image watermark)

AntiFuzz: Impeding Fuzzing Audits of Binary Executables

Repository for tackling Kaggle Ultrasound Nerve Segmentation challenge using Torchnet.

Code for generating the figures in the paper "Capacity of Group-invariant Linear Readouts from Equivariant Representations: How Many Objects can be Linearly Classified Under All Possible Views?"

Minimal diffusion models - Minimal code and simple experiments to play with Denoising Diffusion Probabilistic Models (DDPMs)

Multi-View Consistent Generative Adversarial Networks for 3D-aware Image Synthesis (CVPR2022)

Ejemplo Algoritmo Viterbi - Example of a Viterbi algorithm applied to a hidden Markov model on DNA sequence

Efficiently Disentangle Causal Representations

3D-CariGAN: An End-to-End Solution to 3D Caricature Generation from Normal Face Photos

My 1st place solution at Kaggle Hotel-ID 2021

Code and datasets for the paper "KnowPrompt: Knowledge-aware Prompt-tuning with Synergistic Optimization for Relation Extraction"

OMNIVORE is a single vision model for many different visual modalities

Unified tracking framework with a single appearance model