You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
@@ -206,6 +206,7 @@ A few keywords in `createPotential` would impact the behavior of the function:
206
206
*`const_vals`: the corresponding constraint valu on each residue.
207
207
208
208
*`ethresh`: the accuracy of PME, used to setup the size of the meshgrid of PME.
209
+
*`part_const`: bool type, whether there are constraint groups or not.
209
210
*`has_aux`: bool type, whether to intruduce auxilliary parameters, equilibrated atomic charges, etc.
210
211
211
212
## 3. ADMPQeqGenerator Doc
@@ -252,6 +253,9 @@ The backend of the ADMP QEQ energy is an `ADMPQeqForce` object. It contains the
252
253
*`e_sr`: short range damping energy function
253
254
*`e_site`: on site energy function
254
255
*`coul_energy`: point charge interaction energy function
256
+
*`icount`: times to run get_energy function
257
+
*`hessinv_stride`: strides to update charges by matrix inversion method
258
+
*`qupdate_stride`: strides to update charges
255
259
256
260
***METHOS***
257
261
@@ -278,8 +282,6 @@ The backend of the ADMP QEQ energy is an `ADMPQeqForce` object. It contains the
278
282
Np: interacting pair indices within cutoff distance
279
283
eta:
280
284
N: the atomic length damping parameter
281
-
ds:
282
-
Np: topology distance of pairs
283
285
buffer_scales:
284
286
N * 3: buffer scales on pairs
285
287
mscales:
@@ -288,14 +290,14 @@ The backend of the ADMP QEQ energy is an `ADMPQeqForce` object. It contains the
288
290
energy: charge related coulombic energy plus constraint energy
289
291
```
290
292
291
-
*`E_grads`:
292
-
293
-
Same as `E_full`, only return the concatenation of gradients of funtion `E_ful` on the first two paramters.
294
-
295
-
*`get_energy`:
293
+
*`E_hession`:
294
+
```
295
+
Same as `E_full`, only return hession matrix of `E_full`.
296
+
```
296
297
298
+
*`get_int_energy`:
297
299
```
298
-
This contains charge equilbrant process, and retruns total energy after charge equilibration.
300
+
This contains charge equilbrant process by matrix inversion method, and retruns total energy after charge equilibration.
299
301
300
302
Input:
301
303
positions:
@@ -318,3 +320,30 @@ The backend of the ADMP QEQ energy is an `ADMPQeqForce` object. It contains the
318
320
energy: total energy after charge equilibration
319
321
aux: auxilliary parameters, including equilibrated atomic charges and lagmt
320
322
```
323
+
324
+
*`get_proj_grad`:
325
+
```
326
+
Function to get the value and projected gradients.
327
+
328
+
Input:
329
+
func:
330
+
function: position
331
+
constriant_matrix:
332
+
N * Na: N constrained groups
333
+
has_aux:
334
+
bool type: whether to return auxiliaries.
335
+
```
336
+
337
+
*`get_step_energy`:
338
+
```
339
+
This contains charge equilbrant process by matrix inversion method or projected gradient method, it depends on step count and `hessinv_stride`, and retruns total energy after charge equilibration.
340
+
341
+
The inputs are the as `E_full`.
342
+
```
343
+
344
+
*`get_energy`:
345
+
```
346
+
This contains charge equilbrant process and total energy calculation, retruns total energy every step and update charges by every strides of `qupdate_stride`.
0 commit comments