Skip to content

Remove ase.Atoms dependence of SphericalExpansion by Structure object#3

Open
agoscinski wants to merge 2 commits intorm-rcutfrom
neighbor-matrix
Open

Remove ase.Atoms dependence of SphericalExpansion by Structure object#3
agoscinski wants to merge 2 commits intorm-rcutfrom
neighbor-matrix

Conversation

@agoscinski
Copy link
Owner

@agoscinski agoscinski commented Jun 4, 2023

  • replace Structure class with a Dict[str, torch.Tensor] because torch
    does not support non torch Tensor types for tracing
  • the conversion from ase.Atoms to Dict[str, torch.Tensor] is done within a function
    because TorchScript does not support foreign types
  • add typing for various inputs

@agoscinski agoscinski changed the title cRemove ase.Atoms dependence of Structures class and rename to AtomicStructures Remove ase.Atoms dependence of Structures class and rename to AtomicStructures Jun 4, 2023
@agoscinski agoscinski changed the title Remove ase.Atoms dependence of Structures class and rename to AtomicStructures Remove ase.Atoms dependence of SphericalExpansion by Structure object Jun 4, 2023
@agoscinski agoscinski force-pushed the neighbor-matrix branch 4 times, most recently from 85be8f5 to 20faaee Compare June 19, 2023 15:31
We will change the dtypes of torch tensors to the default type which is
usually 32 bit. To ignore the accuracy loss of the tests, we change
equal to allclose
* change forward signature to a Dict[str, torch.Tensor] because torch
  does not support non torch Tensor types for compiling
* the conversion from ase.Atoms to Dict[str, torch.Tensor] is done within a function
  outside of SphericalExpansion because TorchScript does not support foreign types
* atomic_species is now flattened tensor to consider structures of different size
* add typing for various inputs
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant