xnn.common.models.fast.set_use_fast#

xnn.common.models.fast.set_use_fast(model, use_fast)[source]#

Set use_fast on every model inside model that has fast paths.

model may be a wrapper (forces, dispersion, long-range): the setting reaches every potential inside it with a set_use_fast method.

Returns:

model.

Return type:

torch.nn.Module

Parameters: