Lambda网络-Pytorch
λ网络的实现,这是ImageNet上达到SOTA的一种新的图像识别方法。 新方法利用λ层,该层通过将上下文转换为称为lambda的线性函数并将这些线性函数分别应用于每个输入来捕获交互。
安装
$ pip install lambda-networks
用法
全球背景
import torch
from lambda_networks import LambdaLayer
layer = LambdaLayer (
dim = 32 , # channels going in
dim_out = 32 , # channels
1