Closed nbren12 closed 6 months ago
Wasnt able to reproduce with Modulus on my machine:
>>> import modulus
>>> modulus.__version__
'0.5.0'
>>> from modulus.models.afno import AFNO
>>> model = AFNO.from_checkpoint("fcn.mdlus")
>>> model
AFNO(
(patch_embed): PatchEmbed(
(proj): Conv2d(26, 768, kernel_size=(8, 8), stride=(8, 8))
)
(pos_drop): Dropout(p=0.0, inplace=False)
(blocks): ModuleList(
(0-11): 12 x Block(
(norm1): LayerNorm((768,), eps=1e-06, elementwise_affine=True)
(filter): AFNO2DLayer()
(norm2): LayerNorm((768,), eps=1e-06, elementwise_affine=True)
(mlp): AFNOMlp(
(fc1): Linear(in_features=768, out_features=3072, bias=True)
(act): GELU(approximate='none')
(fc2): Linear(in_features=3072, out_features=768, bias=True)
(drop): Dropout(p=0.0, inplace=False)
)
)
)
(head): Linear(in_features=768, out_features=1664, bias=False)
)
>>>
Could it be you have an old AFNO checkpoint in a model registry? I updated to a new version (auto download should pull this one) after a small API change occurred in Modulus. The v0.2 model has the fix for the error you're seeing I believe.
This got updated I think in 0.4.0 of modulus, interesting its just showing up now.
Version
main
On which installation method(s) does this occur?
No response
Describe the issue
Seems the checkpoint is not compatible with the newer versions of modulus:
Here is the error message
Environment details