core.scripts.fit_normalizers

core.scripts.fit_normalizers#

Copyright (c) Meta, Inc. and its affiliates.

This source code is licensed under the MIT license found in the LICENSE file in the root directory of this source tree.

Attributes#

parser

Functions#

fit_norms(→ None)

Fit dataset mean and std using the standard config

Module Contents#

core.scripts.fit_normalizers.fit_norms(config: dict, output_path: str | pathlib.Path, linref_file: str | pathlib.Path | None = None, linref_target: str = 'energy') → None#

Fit dataset mean and std using the standard config

Parameters:

  • config – config

  • output_path – output path

  • linref_file – path to fitted linear references. IF these are used in training they must be used to compute mean/std

  • linref_target – target using linear references, basically always energy.

core.scripts.fit_normalizers.parser#
Contents