LambdaResNet implementation in TensorFlow 2.2
Project description
LambdaResNet - TensorFlow
All credit goes to lucidrains for sharing the initial pytorch implementation. This repo takes majority of contents from his original repo here.
Implementation of λ-ResNet.
Install
$ pip install tf-lambda-resnet
Usage
LambdaResNet uses localized context as explained in the video.
import tensorflow as tf
from tf_lambda_resnet import LambdaResnet18
model = LambdaResnet18()
out = model(tf.ones([10, 224, 224, 3]))
print(out)
Credits
All the code is taken from the below resources. I have assembled them in a proper manner to make it work for TF2.2
References
Thanks to PistonY for sharing a gist here - lambda_net.py
Citations
@inproceedings{
anonymous2021lambdanetworks,
title={LambdaNetworks: Modeling long-range Interactions without Attention},
author={Anonymous},
booktitle={Submitted to International Conference on Learning Representations},
year={2021},
url={https://openreview.net/forum?id=xTJEN-ggl1b},
note={under review}
}
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
tf-lambda-resnet-0.1.3.tar.gz
(4.6 kB
view hashes)
Built Distribution
Close
Hashes for tf_lambda_resnet-0.1.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | dd3d4959776fc8ad6c133db9f763b152bb1ee466b96e91a2af2b5a2aebfa854a |
|
MD5 | f7deaaa50294689ad7ff2efde7d854af |
|
BLAKE2b-256 | bdfdfe73fbbf6b8ac1d99ad082b10905e6971657b01c07676d21cdb65c8e9752 |