leaderj1001 / LambdaNetworks

Implementing Lambda Networks using Pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

LambdaConv Implementation

cmhungsteve opened this issue · comments

in line 44 lambda_p = F.conv3d(values, self.embedding, padding=(0, self.padding, self.padding)) in model.py
Is there any reason using conv3d?
Can it be implemented using conv2d?

Thank you.