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
The prefactor of energy loss at the start of the training. Should be larger than or equal to 0. If set to none-zero value, the energy label should be provided by file energy.npy in each data system. If both start_pref_energy and limit_pref_energy are set to 0, then the energy will be ignored.
The prefactor of force loss at the start of the training. Should be larger than or equal to 0. If set to none-zero value, the force label should be provided by file force.npy in each data system. If both start_pref_force and limit_pref_force are set to 0, then the force will be ignored.
The prefactor of virial loss at the start of the training. Should be larger than or equal to 0. If set to none-zero value, the virial label should be provided by file virial.npy in each data system. If both start_pref_virial and limit_pref_virial are set to 0, then the virial will be ignored.
The prefactor of atom_ener loss at the start of the training. Should be larger than or equal to 0. If set to none-zero value, the atom_ener label should be provided by file atom_ener.npy in each data system. If both start_pref_atom_ener and limit_pref_atom_ener are set to 0, then the atom_ener will be ignored.
The prefactor of atom_ener loss at the limit of the training, Should be larger than or equal to 0. i.e. the training step goes to infinity.
1165
1165
1166
1166
.. _`loss[ener]/relative_f`:
1167
1167
1168
1168
relative_f:
1169
-
|type: ``NoneType`` | ``float``, optional
1169
+
|type: ``float`` | ``NoneType``, optional
1170
1170
|argument path: ``loss[ener]/relative_f``
1171
1171
1172
1172
If provided, relative force error will be used in the loss. The difference of force will be normalized by the magnitude of the force in the label with a shift given by `relative_f`, i.e. DF_i / ( || F || + relative_f ) with DF denoting the difference between prediction and label and || F || denoting the L2 norm of the label.
@@ -1179,15 +1179,15 @@ loss:
1179
1179
.. _`loss[tensor]/pref`:
1180
1180
1181
1181
pref:
1182
-
|type: ``int`` | ``float``
1182
+
|type: ``float`` | ``int``
1183
1183
|argument path: ``loss[tensor]/pref``
1184
1184
1185
1185
The prefactor of the weight of global loss. It should be larger than or equal to 0. If controls the weight of loss corresponding to global label, i.e. 'polarizability.npy` or `dipole.npy`, whose shape should be #frames x [9 or 3]. If it's larger than 0.0, this npy should be included.
1186
1186
1187
1187
.. _`loss[tensor]/pref_atomic`:
1188
1188
1189
1189
pref_atomic:
1190
-
|type: ``int`` | ``float``
1190
+
|type: ``float`` | ``int``
1191
1191
|argument path: ``loss[tensor]/pref_atomic``
1192
1192
1193
1193
The prefactor of the weight of atomic loss. It should be larger than or equal to 0. If controls the weight of loss corresponding to atomic label, i.e. `atomic_polarizability.npy` or `atomic_dipole.npy`, whose shape should be #frames x ([9 or 3] x #selected atoms). If it's larger than 0.0, this npy should be included. Both `pref` and `pref_atomic` should be provided, and either can be set to 0.0.
0 commit comments