revert_default_dtype¶
- atomate2.forcefields.utils.revert_default_dtype()[source]¶
Context manager for torch.default_dtype.
Reverts it to whatever torch.get_default_dtype() was when entering the context.
Originally added for use with MACE(Relax|Static)Maker. https://github.com/ACEsuit/mace/issues/328
- Return type:
Generator[None, None, None]