Skip to content

Forward api#6

Open
agoscinski wants to merge 36 commits intomasterfrom
forward-api
Open

Forward api#6
agoscinski wants to merge 36 commits intomasterfrom
forward-api

Conversation

@agoscinski
Copy link
Owner

No description provided.

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
* the equistore version was pinned with the wrong character which
  resulted in usage of most recent version when installing
agoscinski and others added 8 commits July 14, 2023 18:58
* add asserts to the allclose statements that got lost
* set combination matrix in computation of alchemical spherical expansion
  because seed cannot recreate the reference combination matrix
* ase.neighborlist can have different order on different systems for
unknown reasons so we need to sort the output according to the metadata
before comparing
* the artificial dataset covers cases
  - perodic images
  - noncubic lattice parameters
  - heterogeneous distribution of species
  - heterogeneous distribution of structure size
  - heterogeneous distribution of lattice parameters

* test have been added for multiple frames using artiifical dataset
* to reduce the number of neighbors we increased the cell size
* to also include higher l channels than 0 and 1 we increased the E_max
  parameter
@agoscinski agoscinski force-pushed the forward-api branch 2 times, most recently from 95289c1 to 514db1d Compare July 19, 2023 14:51
* Spherical and VectorExpansion now accepts direction vectors and
  corresponding metainformation
* created temporary dateset and dataloader for creation of neighborlist
* test_finite_differences is now using artificial dataset
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.

2 participants