Make checkpoint files smaller

[pfebrer]

Checkpoint files currently are much larger than needed because they contain huge arrays for the EquivariantProductBasisBlock modules, which could be recomputed instead of stored.

[peterbjorgensen]

The most robust way would be to modify this line of mace
https://github.com/ACEsuit/mace/blob/44b6da4a5edaa4c3ef867a11728555403d1d475d/mace/modules/symmetric_contraction.py#L110
with the keyword persistent=False to avoid saving the U matrices in the state dict.

Otherwise we would need to hack it something like this https://discuss.pytorch.org/t/how-to-customize-the-module-state-dict-load-state-dict/139604/2

I can try to make a pull request for mace. I can't see any benefit of saving the U matrices for them either.

[pfebrer]

I can't see any benefit of saving the U matrices for them either.

Ok, then I think it makes sense to propose a change upstream!

[pfebrer]

I submitted an issue: ACEsuit/mace#124