Commit e658fc77 authored by Andreas EHRMANN's avatar Andreas EHRMANN
Browse files

Update common_utils/dneb.py

parent b4d53ed8
......@@ -183,8 +183,8 @@ def setup_endpoint_minimization(energy_fn,
""" Independently minimize a collection of systems. In practice this will
be DNEB endpoints.
Args:
Rs: array of initial positions of shape (n, N, dimension), where n is number of
systems, N is the system size, and dimension is the dimension of space.
Rs: array of initial positions of shape (n, N, d), where n is number of
systems, N is the system size, and d is the dimension of space.
Return:
- array of final positions with the same shape.
- maxgrad
......@@ -288,7 +288,7 @@ def find_DNEB_paths(R_endpoints,
This is the primary API function.
Args:
R_endpoints: array of shape (e, N, dimension) corresponding to e endpoints
R_endpoints: array of shape (e, N, d) corresponding to e endpoints
transitions: array of shape (t, 2) where t is the number of transitions to consider.
If e==2, set transitions to [[0,1]] since there is only 1 possible transition.
true_energy_fn: A function that takes an (N,d) array of positions and
......@@ -299,7 +299,7 @@ def find_DNEB_paths(R_endpoints,
Returns:
- array of shape (e, N, dimension) corresponding to the minimized endpoints
- array of shape (e, N, d) corresponding to the minimized endpoints
- list of length t containing DNEB results
"""
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment