Skip to content

yanghu819/Vision_Delta_net

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 

Repository files navigation

VisionDeltaNet is a lightweight vision backbone inspired by Parallelizing Linear Transformers with the Delta Rule over Sequence Length. It combines transformers with convolutions for efficient feature extraction and achieves 70%+ accuracy on CIFAR-10 within 10 epochs.

  1. Install dependencies: pip install torch torchvision.

  2. To train and evaluate on CIFAR-10, run:

python vdelta.py

About

A vision backbone inspired by "Parallelizing Linear Transformers with the Delta Rule over Sequence Length" https://arxiv.org/pdf/2406.06484

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages