code questions about Transformer3DBlock #1319
Replies: 1 comment
-
|
This is not an issue with the current codebase. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
physicsnemo/nn/transformer_layers.py line144：
shifted_x = torch.roll(x, shifts=(-shift_pl, -shift_lat, -shift_lat), dims=(1, 2, 3))
I'm not sure if it‘s right. Why not shifted_x = torch.roll(x, shifts=(-shift_pl, -shift_lat, -shift_lon), dims=(1, 2, 3))?
Beta Was this translation helpful? Give feedback.
All reactions