All notable changes to this project will be documented in this file.
The format is based on Keep a Changelog, and this project adheres to Semantic Versioning.
Most recent change on the bottom.
to_ase
method inAtomicData.py
to convertAtomicData
object to (list of)ase.Atoms
object(s)SequentialGraphNetwork
now has insertion methodsnn.SaveForOutput
nequip-evaluate
command for evaluating (metrics on) trained modelsAtomicData.from_ase
now catchesenergy
/energies
arrays
- Nonlinearities now specified with
e
ando
instead of1
and-1
- Update interfaces for
torch_geometric
1.7.1 ande3nn
0.3.3 nonlinearity_scalars
now also affects the nonlinearity used in the radial net ofInteractionBlock
- Cleaned up naming of initializers
- Fix specifying nonlinearities when wandb enabled
Final
backport for <3.8 compatability- Fixed
nequip-*
commands when usingpip install
- Default models rescale per-atom energies, and not just total
- Fixed Python <3.8 backward compatability with
atomic_save
- Option for which nonlinearities to use
- Option to save models every n epochs in training
- Option to specify optimization defaults for
e3nn
- Using
wandb
no longer breaks the inclusion of special objects like callables in configs
iepoch
is no longer off-by-one when restarting a training run that hitmax_epochs
- Builders, and not just sub-builders, use the class name as a default prefix
early_stopping_xxx
arguments added to enable early stop for platued values or values that out of lower/upper bounds.
- Sub-builders can be skipped in
instantiate
by setting them toNone
- More flexible model initialization
- Add MD w/ Nequip-ASE-calculator + run-MD script w/ custom Nose-Hoover
- PBC must be explicit if a cell is provided
- Training now uses atomic file writes to avoid corruption if interupted
feature_embedding
renamed tochemical_embedding
in default models
BesselBasis
now works on GPU whentrainable=False
- Dataset
extra_fixed_fields
are now added even ifget_data()
returnsAtomicData
objects
load_deployed_model
now correctly loads all metadata