AFNO-transformer icon indicating copy to clipboard operation
AFNO-transformer copied to clipboard

Published 20 hours ago • verified publisher icon NVlabs

Fixed dimension of frequency threshold in 2d

Open scaomath opened this issue 2 years ago • 0 comments

Based on the paper's Figure 3: https://openreview.net/pdf?id=EXHG-A3jlM

I think the number of modes threshold in 2d https://github.com/NVlabs/AFNO-transformer/blob/941e924ffdb34cb93897b5fd6e22e5965fdc1111/afno/afno2d.py#L54-L55 should be W instead of N since N = H*W, similar applies to bfno2d.py.

Nice paper btw. @jtguibas

scaomath avatar Nov 07 '22 00:11 scaomath