lambda-networks
lambda-networks copied to clipboard
Image Size
Are non-square image blocks allowed for context? Using global context and a non-square dimensions (96, 128), I get an error on this line about dimension size.
λp = einsum('n m k u, b u v m -> b n k v', rel_pos_emb, v)