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