easonyang1996 / LambdaResnet

Pytorch implementation of LambdaResnet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

LambdaResnet

Pytorch implementation of LambdaResnet

Statement

lambda_layer.py is implementated by lucidrains(https://github.com/lucidrains/lambda-networks).

lambda_resnet.py is implementated based on the official implementation of Resnet by Torchvision.

Citation

@inproceedings{
bello2021lambdanetworks,
title={LambdaNetworks: Modeling long-range Interactions without Attention},
author={Irwan Bello},
booktitle={International Conference on Learning Representations},
year={2021},
url={https://openreview.net/forum?id=xTJEN-ggl1b}
}

About

Pytorch implementation of LambdaResnet

License:MIT License


Languages

Language:Python 100.0%